✅ conform to new testing data structure (#87)

Author: ianardee

mindee/documents/base.py

@@ -34,7 +34,7 @@ def __init__(
         self,
         input_file: InputDocument,
         document_type: str,
-        api_prediction: dict,
+        api_prediction: TypeApiPrediction,
         page_n: Optional[int] = None,
     ):
         if input_file:

mindee/documents/custom_document.py

@@ -10,7 +10,7 @@ class CustomDocument(Document):
     def __init__(
         self,
         document_type: str,
-        api_prediction=None,
+        api_prediction: TypeApiPrediction,
         input_file=None,
         page_n: Optional[int] = None,
     ):
@@ -51,7 +51,7 @@ def __str__(self) -> str:
                 name,
                 " ".join([value["content"] for value in info["values"]]),
             )
-        custom_doc_str += "-----------------\n"
+        custom_doc_str += "----------------------"
         return custom_doc_str
 
     def _checklist(self) -> None:

tests/__init__.py

@@ -0,0 +1,4 @@
+RECEIPT_DATA_DIR = "./tests/data/receipt"
+INVOICE_DATA_DIR = "./tests/data/invoice"
+PASSPORT_DATA_DIR = "./tests/data/passport"
+CUSTOM_DATA_DIR = "./tests/data/custom"