🐛 fix EU imports, use short imports in tests (#128)

ianardee · web-flow · commit fd7f40673290 · 2023-04-20T17:39:05.000+02:00
diff --git a/mindee/documents/__init__.py b/mindee/documents/__init__.py
@@ -1,7 +1,6 @@
-from mindee.documents import fr, us
+from mindee.documents import eu, fr, us
 from mindee.documents.cropper import CropperV1, TypeCropperV1
 from mindee.documents.custom import CustomV1, TypeCustomV1
-from mindee.documents.eu import LicensePlateV1, TypeLicensePlateV1
 from mindee.documents.financial import (
     FinancialDocumentV1,
     FinancialV1,
diff --git a/tests/documents/eu/test_license_plate_v1.py b/tests/documents/eu/test_license_plate_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.eu.license_plate.license_plate_v1 import LicensePlateV1
+from mindee.documents.eu import LicensePlateV1
 
 EU_LICENSE_PLATE_DATA_DIR = "./tests/data/eu/license_plate"
 
diff --git a/tests/documents/fr/test_bank_account_details_v1.py b/tests/documents/fr/test_bank_account_details_v1.py
@@ -2,9 +2,7 @@
 
 import pytest
 
-from mindee.documents.fr.bank_account_details.bank_account_details_v1 import (
-    BankAccountDetailsV1,
-)
+from mindee.documents.fr import BankAccountDetailsV1
 
 FR_BANK_ACCOUNT_DETAILS_DATA_DIR = "./tests/data/fr/bank_account_details"
 
diff --git a/tests/documents/fr/test_carte_grise_v1.py b/tests/documents/fr/test_carte_grise_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.fr.carte_grise.carte_grise_v1 import CarteGriseV1
+from mindee.documents.fr import CarteGriseV1
 from tests import FR_CARTE_GRISE_DATA_DIR
 
 FILE_PATH_FR_CARTE_GRISE_V1_COMPLETE = (
diff --git a/tests/documents/fr/test_carte_vitale_v1.py b/tests/documents/fr/test_carte_vitale_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.fr.carte_vitale.carte_vitale_v1 import CarteVitaleV1
+from mindee.documents.fr import CarteVitaleV1
 
 FR_CARTE_VITALE_DATA_DIR = "./tests/data/fr/carte_vitale"
 
diff --git a/tests/documents/fr/test_id_card_v1.py b/tests/documents/fr/test_id_card_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.fr.id_card.id_card_v1 import IdCardV1
+from mindee.documents.fr import IdCardV1
 
 FR_ID_CARD_DATA_DIR = "./tests/data/fr/id_card"
 
diff --git a/tests/documents/test_cropper_v1.py b/tests/documents/test_cropper_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.cropper.cropper_v1 import CropperV1
+from mindee.documents import CropperV1
 from tests import CROPPER_DATA_DIR
 
 FILE_PATH_CROPPER_V1_COMPLETE = f"{CROPPER_DATA_DIR}/response_v1/complete.json"
diff --git a/tests/documents/test_custom_v1.py b/tests/documents/test_custom_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.custom.custom_v1 import CustomV1
+from mindee.documents import CustomV1
 from mindee.fields.api_builder import ClassificationField, ListField, ListFieldValue
 from tests import CUSTOM_DATA_DIR
 
diff --git a/tests/documents/test_financial_document_v1.py b/tests/documents/test_financial_document_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.financial.financial_document_v1 import FinancialDocumentV1
+from mindee.documents import FinancialDocumentV1
 from tests import FINANCIAL_DOC_DATA_DIR
 
 FILE_PATH_FINANCIAL_DOC_V1_INVOICE = (
diff --git a/tests/documents/test_financial_v1.py b/tests/documents/test_financial_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.financial.financial_v1 import FinancialV1
+from mindee.documents import FinancialV1
 from tests.documents.test_invoice_v3 import (
     FILE_PATH_INVOICE_V3_COMPLETE,
     FILE_PATH_INVOICE_V3_EMPTY,
diff --git a/tests/documents/test_passport_v1.py b/tests/documents/test_passport_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.passport.passport_v1 import PassportV1
+from mindee.documents import PassportV1
 from tests import PASSPORT_DATA_DIR
 
 FILE_PATH_PASSPORT_V1_COMPLETE = f"{PASSPORT_DATA_DIR}/response_v1/complete.json"
diff --git a/tests/documents/test_proof_of_address_v1.py b/tests/documents/test_proof_of_address_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.proof_of_address.proof_of_address_v1 import ProofOfAddressV1
+from mindee.documents import ProofOfAddressV1
 
 PROOF_OF_ADDRESS_DATA_DIR = "./tests/data/proof_of_address"
 
diff --git a/tests/documents/test_receipt_v3.py b/tests/documents/test_receipt_v3.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.receipt.receipt_v3 import ReceiptV3
+from mindee.documents import ReceiptV3
 from tests import RECEIPT_DATA_DIR
 
 FILE_PATH_RECEIPT_V3_COMPLETE = f"{RECEIPT_DATA_DIR}/response_v3/complete.json"
diff --git a/tests/documents/test_receipt_v4.py b/tests/documents/test_receipt_v4.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.receipt.receipt_v4 import ReceiptV4
+from mindee.documents import ReceiptV4
 from tests import RECEIPT_DATA_DIR
 
 FILE_PATH_RECEIPT_V4_COMPLETE = f"{RECEIPT_DATA_DIR}/response_v4/complete.json"
diff --git a/tests/documents/test_receipt_v5.py b/tests/documents/test_receipt_v5.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.receipt.receipt_v5 import ReceiptV5
+from mindee.documents import ReceiptV5
 from tests import RECEIPT_DATA_DIR
 
 FILE_PATH_RECEIPT_V5_COMPLETE = f"{RECEIPT_DATA_DIR}/response_v5/complete.json"
diff --git a/tests/documents/test_shipping_container_v1.py b/tests/documents/test_shipping_container_v1.py
@@ -2,9 +2,7 @@
 
 import pytest
 
-from mindee.documents.shipping_container.shipping_container_v1 import (
-    ShippingContainerV1,
-)
+from mindee.documents import ShippingContainerV1
 
 SHIPPING_CONTAINER_DATA_DIR = "./tests/data/shipping_container"
 
diff --git a/tests/documents/us/test_bank_check_v1.py b/tests/documents/us/test_bank_check_v1.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from mindee.documents.us.bank_check.bank_check_v1 import BankCheckV1
+from mindee.documents.us import BankCheckV1
 from tests import US_BANK_CHECK_DATA_DIR
 
 FILE_PATH_US_BANK_CHECK_V1_COMPLETE = (
