Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion mindee/parsing/v2/inference_result_fields.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
2 changes: 1 addition & 1 deletion mindee/parsing/v2/simple_field.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 "")
18 changes: 15 additions & 3 deletions tests/v2/test_inference_response.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
)
Expand Down