Begin implementation, basic inferenece

test/util_test.py

@@ -0,0 +1,17 @@
+import unittest
+
+from . import test_data
+
+from sia import util
+
+class UtilTest(unittest.TestCase):
+    def test_get_valid_root_elements_single(self):
+        valid_elements = util.get_valid_root_elements(test_data.echo_action_schema)
+        self.assertEqual(valid_elements, {'test_tag'})
+
+    def test_split_response_single_element(self):
+        response = "Some reasoning here\n<test_tag>content</test_tag>"
+        valid_elements = {'test_tag'}
+        result = util.split_response(response, valid_elements)
+        self.assertEqual(result.reasoning, "Some reasoning here")
+        self.assertEqual(result.actions, "<test_tag>content</test_tag>")