diff --git a/mindee/parsing/v2/inference_result_fields.py b/mindee/parsing/v2/inference_result_fields.py index f30b30c6..5ffbfba9 100644 --- a/mindee/parsing/v2/inference_result_fields.py +++ b/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 diff --git a/mindee/parsing/v2/simple_field.py b/mindee/parsing/v2/simple_field.py index 82445544..4e0ee4cb 100644 --- a/mindee/parsing/v2/simple_field.py +++ b/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 "") diff --git a/tests/data b/tests/data index a6b8e649..5e9068a7 160000 --- a/tests/data +++ b/tests/data @@ -1 +1 @@ -Subproject commit a6b8e649700413e6b32d886c1362eb6c75e094e8 +Subproject commit 5e9068a7eaed322d0738212ea624db7aa00e9a76 diff --git a/tests/v2/test_inference_response.py b/tests/v2/test_inference_response.py index c5e32f6a..55ddb3f6 100644 --- a/tests/v2/test_inference_response.py +++ b/tests/v2/test_inference_response.py @@ -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 )