🐛 Fix financial doc

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Mindee python SDK
 
+## v1.1.2 (2020-02-19)
+
+### Fix
+
+* :bug: Fixed FinancialDoc invoice version and reconstruction
+
 ## v1.1.1 (2020-01-31)
 
 ### Chg

README.md

@@ -1,4 +1,4 @@
-# Mindee API helper library for python
+# Receipt, passport and invoice OCR python helper for Mindee API
 
 The full documentation is available [here](https://mindee.com/documentation/get-started/setup-your-account)
 

mindee/documents/financial_document.py

@@ -192,7 +192,7 @@ def request(
         :param invoice_token: Invoices API token
         """
         if "pdf" in input_file.file_extension:
-            url = os.path.join(base_url, "invoices", "v1", "predict")
+            url = os.path.join(base_url, "invoices", "v2", "predict")
             return request(url, input_file, invoice_token, include_words)
         else:
             url = os.path.join(base_url, "expense_receipts", "v3", "predict")
@@ -218,8 +218,9 @@ def __taxes_match_total_incl(self):
         total_vat = 0
         reconstructed_total = 0
         for tax in self.taxes:
-            total_vat += tax.value
-            reconstructed_total += tax.value + 100 * tax.value / tax.rate
+            if tax.rate is not None and tax.rate != 0:
+                total_vat += tax.value
+                reconstructed_total += tax.value + 100 * tax.value / tax.rate
 
         # Sanity check
         if total_vat <= 0:

requirements.dev.txt

@@ -5,4 +5,5 @@ pytz~=2020.1
 setuptools~=49.2.0
 matplotlib~=3.1.2
 numpy~=1.18.5
-PyMuPDF~=1.18.1
+PyMuPDF~=1.18.1
+pytest-cov~=2.11.1

tests/documents/test_financial_doc.py

@@ -236,6 +236,14 @@ def test__invoice_taxes_match_total_incl_3(invoice_pred):
     assert financial_doc.checklist["taxes_match_total_incl"] is False
 
 
+def test__shouldnt_raise_when_tax_rate_none(invoice_pred):
+    # sanity check with null tax
+    invoice_pred["total_incl"] = {"value": 507.25, "probability": 0.6}
+    invoice_pred["taxes"] = [{"rate": "N/A", "value": 0., "probability": 0.5}]
+    financial_doc = FinancialDocument(invoice_pred)
+    assert financial_doc.checklist["taxes_match_total_incl"] is False
+
+
 def test_compare_1(financial_doc_from_invoice_object):
     # Compare same object must return all True
     benchmark = FinancialDocument.compare(financial_doc_from_invoice_object, financial_doc_from_invoice_object)

tests/documents/test_invoice.py

@@ -252,6 +252,15 @@ def test__taxes_plus_total_excl_match_total_incl_3(invoice_pred):
     assert invoice.checklist["taxes_match_total_incl"] is False
 
 
+def test__shouldnt_raise_when_tax_rate_none(invoice_pred):
+    # sanity check with null tax
+    invoice_pred["total_excl"] = {"value": 456.15, "probability": 0.6}
+    invoice_pred["total_incl"] = {"value": 507.25, "probability": 0.6}
+    invoice_pred["taxes"] = [{"rate": "N/A", "value": 0., "probability": 0.5}]
+    invoice = Invoice(invoice_pred)
+    assert invoice.checklist["taxes_match_total_incl"] is False
+
+
 def test_compare_1(invoice_object):
     # Compare same object must return all True
     benchmark = Invoice.compare(invoice_object, invoice_object)