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
4 changes: 4 additions & 0 deletions mindee/v2/parsing/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,9 @@
from mindee.v2.parsing.error.error_response import ErrorResponse
from mindee.v2.parsing.inference.base_inference import BaseInference
from mindee.v2.parsing.inference.base_response import BaseResponse
from mindee.v2.parsing.inference.failed_inference_response import (
FailedInferenceResponse,
)
from mindee.v2.parsing.inference.inference_active_options import InferenceActiveOptions
from mindee.v2.parsing.inference.inference_file import InferenceFile
from mindee.v2.parsing.inference.inference_model import InferenceModel
Expand All @@ -18,6 +21,7 @@
"ExtractionInference",
"ExtractionResponse",
"ExtractionResult",
"FailedInferenceResponse",
"InferenceActiveOptions",
"InferenceFile",
"InferenceModel",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

from mindee.parsing.common.common_response import CommonResponse
from mindee.parsing.common.string_dict import StringDict
from mindee.v2.parsing import ErrorResponse
from mindee.v2.parsing.error.error_response import ErrorResponse


class FailedInferenceResponse(CommonResponse):
Expand Down
2 changes: 1 addition & 1 deletion mindee/v2/parsing/inference/field/simple_field.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ def string_value(self) -> str | None:
return self.value

@property
def number_value(self) -> float | None:
def float_value(self) -> float | None:
"""Retrieves a number field value as a float."""
if self.value is not None and not isinstance(self.value, float):
raise ValueError("Value is not a number")
Expand Down
2 changes: 1 addition & 1 deletion tests/v2/parsing/test_failed_inference_response.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

import pytest

from mindee.v2.parsing.failed_inference_response import FailedInferenceResponse
from mindee.v2.parsing import FailedInferenceResponse
from tests.utils import V2_DATA_DIR


Expand Down
12 changes: 6 additions & 6 deletions tests/v2/parsing/test_simple_field.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,27 +32,27 @@ def test_raises_when_value_is_boolean(self):
class TestSimpleFieldNumberValue:
def test_returns_float_when_value_is_float(self):
field = _make_field(3.14)
assert field.number_value == 3.14
assert field.float_value == 3.14

def test_returns_float_when_value_is_int(self):
# Integers are coerced to float in the constructor
field = SimpleField({"value": 42})
assert field.number_value == 42.0
assert isinstance(field.number_value, float)
assert field.float_value == 42.0
assert isinstance(field.float_value, float)

def test_returns_none_when_value_is_none(self):
field = _make_field(None)
assert field.number_value is None
assert field.float_value is None

def test_raises_when_value_is_string(self):
field = _make_field("42")
with pytest.raises(ValueError, match="Value is not a number"):
_ = field.number_value
_ = field.float_value

def test_raises_when_value_is_boolean(self):
field = _make_field(True)
with pytest.raises(ValueError, match="Value is not a number"):
_ = field.number_value
_ = field.float_value


@pytest.mark.v2
Expand Down