add simple test case for build_entities_from_data

tests/test_utils_build_entities_from_data.py

@@ -0,0 +1,68 @@
+import json
+
+from dataintegration.entity import EntitySchema, EntityPath
+from dataintegration.utils import build_entities_from_data
+
+
+def test_single_object():
+    """test write to single object"""
+    test_data = """
+{
+  "name": "sai",
+  "email": "[email protected]"
+}"""
+    data = json.loads(test_data)
+    entities = build_entities_from_data(data)
+    assert len(entities.entities) == 1
+    for _ in entities.entities:
+        assert len(_.values) == 2
+        assert _.values == [["sai"], ["[email protected]"]]
+    assert len(entities.schema.paths) == 2
+    assert str(entities.schema) == str(EntitySchema(
+        type_uri="",
+        paths=[
+            EntityPath("name", False, is_attribute=True),
+            EntityPath("email", False, is_attribute=True),
+        ]
+    ))
+
+
+def test_single_object_one_level():
+    """test write to single object"""
+    test_data = """
+{
+  "name": "sai",
+  "email": "[email protected]",
+  "city": {
+    "name": "San Francisco",
+    "country": "United States"
+  }
+}"""
+    data = json.loads(test_data)
+    entities = build_entities_from_data(data)
+    assert len(entities.entities) == 1
+    for _ in entities.entities:
+        assert len(_.values) == 3
+        assert _.values[0:2] == [["sai"], ["[email protected]"]]
+        assert _.values[2][0].startswith("urn:x-ulid:")
+    assert len(entities.schema.paths) == 3
+    assert str(entities.schema) == str(EntitySchema(
+        type_uri="",
+        paths=[
+            EntityPath("name", False, is_attribute=True),
+            EntityPath("email", False, is_attribute=True),
+            EntityPath("city", True, is_attribute=True),
+        ]
+    ))
+    # Validate sub entities
+    for _ in entities.sub_entities:
+        for _entity in _.entities:
+            assert len(_entity.values) == 2
+            assert _entity.values == [["San Francisco"], ["United States"]]
+        assert str(_.schema) == str(EntitySchema(
+            type_uri="",
+            paths=[
+                EntityPath("name", False, is_attribute=True),
+                EntityPath("country", False, is_attribute=True)
+            ]
+        ))