#16: bindings: Add a bit more functionality to binded function

bindings/python/tv.cc

@@ -5,6 +5,7 @@ namespace vt::tv::bindings::python {
 void tv_from_json(const std::string& input_json_str, const std::string& input_yaml_params_str) {
   auto j = nlohmann::json::parse(input_json_str);
   fmt::print("JSON: {}\n", j.dump(2));
+  YAML::Node config = YAML::Load(input_yaml_params_str);
   // Read the json file
   // using json = nlohmann::json;
 

bindings/python/tv.h

@@ -54,8 +54,9 @@
 #include "vt-tv/utility/input_iterator.h"
 
 #include <nlohmann/json.hpp>
-#include <nanobind/nanobind.h>
+#include <yaml-cpp/yaml.h>
 
+#include <nanobind/nanobind.h>
 #include <nanobind/stl/string.h>
 
 namespace vt::tv::bindings::python {

setup.py

@@ -38,9 +38,6 @@ def build_extension(self, ext):
     self.spawn(['cmake', ext.sourcedir] + cmake_args)
     if not self.dry_run:
       self.spawn(['cmake', '--build', '.', '--parallel', '-j8'] + build_args)
-    # # Returning to the previous directory
-    # print(self.distribution.get_fullname())
-    # os.chdir(self.distribution.get_fullname())
 
 
 setup(

testvttv.py

@@ -0,0 +1,19 @@
+import json
+import yaml
+import vttv
+
+with open('/Users/pierrepebay/Develop/vt-tv/tests/unit/lb_test_data/data.0.json') as f:
+  data = json.load(f)
+
+
+with open("/Users/pierrepebay/Develop/vt-tv/config/conf.yaml", "r") as stream:
+  try:
+    params = yaml.safe_load(stream)
+  except yaml.YAMLError as exc:
+    print(exc)
+
+
+data_serialized = json.dumps(data)
+params_serialized = yaml.dump(params)
+
+vttv.tv_from_json(data_serialized, params_serialized)