Add tests

Author: Val Brodsky

tests/data/annotation_types/test_ner.py

@@ -26,14 +26,8 @@ def test_document_entity():
 
 
 def test_conversation_entity():
-    document_entity = ConversationEntity(name="tool_name",
-                                         message_id=1,
-                                         start=0,
-                                         end=1,
-                                         confidence=0.5)
+    conversation_entity = ConversationEntity(message_id=1, start=0, end=1)
 
-    assert document_entity.name == "tool_name"
-    assert document_entity.message_id == "1"
-    assert document_entity.start == 0
-    assert document_entity.end == 1
-    assert document_entity.confidence == 0.5
+    assert conversation_entity.message_id == "1"
+    assert conversation_entity.start == 0
+    assert conversation_entity.end == 1

tests/data/serialization/ndjson/test_text_entity.py

@@ -15,3 +15,15 @@ def test_text_entity_import(filename: str):
     res = list(NDJsonConverter.deserialize(data))
     res = list(NDJsonConverter.serialize(res))
     assert res == data
+
+
+@pytest.mark.parametrize("filename", [
+    "tests/data/assets/ndjson/conversation_entity_import.json",
+    "tests/data/assets/ndjson/conversation_entity_without_confidence_import.json"
+])
+def test_conversation_entity_import(filename: str):
+    with open(filename, 'r') as file:
+        data = json.load(file)
+    res = list(NDJsonConverter.deserialize(data))
+    res = list(NDJsonConverter.serialize(res))
+    assert res == data