Skip to content

Commit

Permalink
add str & stringfield tests
Browse files Browse the repository at this point in the history
  • Loading branch information
sebastianMindee committed Jan 15, 2024
1 parent fb50e79 commit 6859829
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions tests/product/generated/test_generated_v1.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
from mindee.parsing.generated.generated_list import GeneratedListField
from mindee.parsing.generated.generated_object import GeneratedObjectField
from mindee.parsing.standard.position import PositionField
from mindee.parsing.standard.text import StringField
from mindee.product.generated.generated_v1 import GeneratedV1
from mindee.product.generated.generated_v1_document import GeneratedV1Document
from mindee.product.generated.generated_v1_page import GeneratedV1Page
Expand Down Expand Up @@ -156,6 +157,14 @@ def test_invoice_v4_complete_doc(invoice_v4_complete_doc) -> None:
invoice_v4_complete_doc.inference.prediction.fields["locale"],
GeneratedObjectField,
)
assert isinstance(
invoice_v4_complete_doc.inference.prediction.fields["locale"].currency,
str,
)
assert isinstance(
invoice_v4_complete_doc.inference.prediction.fields["document_type"],
StringField,
)
assert isinstance(
invoice_v4_complete_doc.inference.prediction.fields[
"supplier_company_registrations"
Expand Down

0 comments on commit 6859829

Please sign in to comment.