✅ update v2 inference tests

Author: ianardee

mindee/parsing/v2/inference_result_fields.py

@@ -23,7 +23,10 @@ def __str__(self) -> str:
         str_fields = ""
         for field_key, field_value in self.items():
             if field_value.field_type == FieldType.SIMPLE:
-                str_fields += f"\n:{field_key}: {field_value}"
+                final_value = f"{field_value}"
+                if final_value:
+                    final_value = f" {final_value}"
+                str_fields += f"\n:{field_key}:{final_value}"
             else:
                 str_fields += f"\n:{field_key}:{field_value}"
         return str_fields

mindee/parsing/v2/simple_field.py

@@ -15,4 +15,4 @@ def __init__(self, raw_response: StringDict, indent_level: int = 0):
         self.value = raw_response["value"] = raw_response.get("value", None)
 
     def __str__(self) -> str:
-        return str(self.value)
+        return str(self.value if self.value is not None else "")

tests/data

@@ -112,9 +112,21 @@ def test_standard_field_types():
     json_sample, rst_sample = _get_inference_samples("standard_field_types")
     inference_result = InferenceResponse(json_sample)
     assert isinstance(inference_result.inference, Inference)
-    assert isinstance(
-        inference_result.inference.result.fields.field_simple, SimpleField
-    )
+    field_simple_string = inference_result.inference.result.fields.field_simple_string
+    assert isinstance(field_simple_string, SimpleField)
+    assert field_simple_string.value == "field_simple_string-value"
+    assert str(field_simple_string) == "field_simple_string-value"
+
+    field_simple_bool = inference_result.inference.result.fields.field_simple_bool
+    assert isinstance(field_simple_bool, SimpleField)
+    assert field_simple_bool.value is True
+    assert str(field_simple_bool) == "True"
+
+    field_simple_null = inference_result.inference.result.fields.field_simple_null
+    assert isinstance(field_simple_null, SimpleField)
+    assert field_simple_null.value is None
+    assert str(field_simple_null) == ""
+
     assert isinstance(
         inference_result.inference.result.fields.field_object, ObjectField
     )