def test__taxes_plus_total_excl_match_total_incl_3(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": 20, "value": 0., "probability": 0.5}]
invoice = Invoice(invoice_pred)
assert invoice.checklist["taxes_match_total_incl"] is False
def test__reconstruct_total_incl_from_taxes_plus_excl_2(invoice_pred):
# no excl implies no reconstruct for total incl
invoice_pred["total_incl"] = {"value": "N/A", "probability": 0.}
invoice_pred["total_excl"] = {"value": "N/A", "probability": 0.}
invoice_pred["taxes"] = [{"rate": 20, "value": 9.5, "probability": 0.9}]
invoice = Invoice(invoice_pred)
assert invoice.total_incl.value is None
def test__reconstruct_total_excl_from_tcc_and_taxes_2(invoice_pred):
# no taxes implies no reconstruct for total excl
invoice_pred["total_incl"] = {"value": 1150.20, "probability": 0.7}
invoice_pred["total_excl"] = {"value": "N/A", "probability": 0.}
invoice_pred["taxes"] = []
invoice = Invoice(invoice_pred)
assert invoice.total_excl.value is None
def test_compare_3(invoice_object_from_scratch):
# Compare invoice from class
benchmark = Invoice.compare(invoice_object_from_scratch,
invoice_object_from_scratch)
for key in benchmark.keys():
if "__acc__" in key:
assert benchmark[key] is True
def test__reconstruct_total_incl_from_taxes_plus_excl_4(invoice_pred):
# working example
invoice_pred["total_incl"] = {"value": "N/A", "probability": 0.}
invoice_pred["total_excl"] = {"value": 240.5, "probability": 0.9}
invoice_pred["taxes"] = [{"rate": 20, "value": 9.5, "probability": 0.9}]
invoice = Invoice(invoice_pred)
assert invoice.total_incl.value == 250
assert invoice.total_incl.probability == 0.81
def test__reconstruct_total_incl_from_taxes_plus_excl_3(invoice_pred):
# incl already exists implies no reconstruct
invoice_pred["total_incl"] = {"value": 260, "probability": 0.4}
invoice_pred["total_excl"] = {"value": 240.5, "probability": 0.9}
invoice_pred["taxes"] = [{"rate": 20, "value": 9.5, "probability": 0.9}]
invoice = Invoice(invoice_pred)
assert invoice.total_incl.value == 260
assert invoice.total_incl.probability == 0.4
def test__reconstruct_total_excl_from_tcc_and_taxes_3(invoice_pred):
# excl already exists implies no reconstruct
invoice_pred["total_incl"] = {"value": 1150.20, "probability": 0.7}
invoice_pred["total_excl"] = {"value": 1050.0, "probability": 0.4}
invoice_pred["taxes"] = []
invoice = Invoice(invoice_pred)
assert invoice.total_excl.value == 1050.0
assert invoice.total_excl.probability == 0.4
def test_compare_4(invoice_object_from_scratch):
# Compare invoice from class with empty taxes
invoice_object_from_scratch.taxes = []
benchmark = Invoice.compare(invoice_object_from_scratch,
invoice_object_from_scratch)
for key in benchmark.keys():
if "__acc__" in key:
assert benchmark[key] is True
elif "__pre__" in key:
assert benchmark[key] in [True, None]
def invoice_object_from_scratch():
return Invoice(locale="fr",
total_incl=12,
total_excl=15,
invoice_date="2018-12-21",
invoice_number="001",
due_date="2019-01-01",
taxes={(1, 10), (2, 20)},
supplier="Amazon",
payment_details="1231456498799765",
company_number="asdqsdae",
orientation=0,
total_tax=3)
def test__taxes_match_total_excl_2(invoice_pred):
# not matching example with close error
invoice_pred["total_excl"] = {"value": 456.15, "probability": 0.6}
invoice_pred["taxes"] = [{
"rate": 20,
"value": 10.9,
"probability": 0.5
}, {
"rate": 10,
"value": 40.12,
"probability": 0.1
}]
invoice = Invoice(invoice_pred)
assert invoice.checklist["taxes_match_total_excl"] is False
def test__reconstruct_total_tax_2(invoice_pred):
# working example
invoice_pred["taxes"] = [{
"rate": 20,
"value": 10.2,
"probability": 0.5
}, {
"rate": 10,
"value": 40.0,
"probability": 0.1
}]
invoice = Invoice(invoice_pred)
assert invoice.total_tax.value == 50.2
assert invoice.total_tax.probability == 0.05
def test_null_tax_rates_dont_raise():
invoice = Invoice(locale="fr",
total_incl=12,
total_excl=15,
invoice_date="2018-12-21",
invoice_number="001",
due_date="2019-01-01",
taxes={(1, 0), (2, 20)},
supplier="Amazon",
payment_details="1231456498799765",
company_number="asdqsdae",
orientation=0,
total_tax=3)
assert invoice.checklist["taxes_match_total_incl"] is False
assert invoice.checklist["taxes_match_total_excl"] is False
def test__reconstruct_total_excl_from_tcc_and_taxes_4(invoice_pred):
# working example
invoice_pred["total_incl"] = {"value": 1150.20, "probability": 0.6}
invoice_pred["total_excl"] = {"value": "N/A", "probability": 0.}
invoice_pred["taxes"] = [{
"rate": 20,
"value": 10.2,
"probability": 0.5
}, {
"rate": 10,
"value": 40.0,
"probability": 0.1
}]
invoice = Invoice(invoice_pred)
assert invoice.total_excl.value == 1100
assert invoice.total_excl.probability == 0.03
def test__taxes_match_total_excl_1(invoice_pred):
# matching example
invoice_pred["total_excl"] = {"value": 456.15, "probability": 0.6}
invoice_pred["taxes"] = [{
"rate": 20,
"value": 10.99,
"probability": 0.5
}, {
"rate": 10,
"value": 40.12,
"probability": 0.1
}]
invoice = Invoice(invoice_pred)
assert invoice.checklist["taxes_match_total_excl"] is True
assert invoice.total_excl.probability == 1.
for tax in invoice.taxes:
assert tax.probability == 1.
def invoice_object():
json_repsonse = json.load(open("./tests/data/invoices/v2/invoice.json"))
return Invoice(json_repsonse["predictions"][0])
def test_empty_object_works():
invoice = Invoice()
assert invoice.total_tax.value is None
def test_compare_2(invoice_object, invoice_object_all_na):
# Compare full object and empty object
benchmark = Invoice.compare(invoice_object, invoice_object_all_na)
for key in benchmark.keys():
assert benchmark[key] is False
def test__reconstruct_total_tax_1(invoice_pred):
# no taxes implies no reconstruct for total tax
invoice_pred["taxes"] = []
invoice = Invoice(invoice_pred)
assert invoice.total_tax.value is None
def test_compare_1(invoice_object):
# Compare same object must return all True
benchmark = Invoice.compare(invoice_object, invoice_object)
for value in benchmark.values():
assert value is True
# Invoice tests
def test_responseWrapper_invoice(dummy_file_input):
response = json.load(open('./tests/data/invoices/v2/invoice.json'))
parsed_invoice = Response.format_response(response, "invoice",
dummy_file_input)
assert parsed_invoice.invoice.invoice_date.value == "2020-02-17"
from mindee import Invoice
json_repsonse = json.load(open("./tests/data/invoices/v2/invoice.json"))
invoice_0 = Invoice(json_repsonse["predictions"][0])
invoice_1 = Invoice(json_repsonse["predictions"][1])
def test_merge_pages():
merge_document = Document.merge_pages([invoice_0, invoice_1])
assert merge_document.invoice_date.value == "2020-02-17"
# Receipt tests
def test_responseWrapper_receipt(dummy_file_input):
response = json.load(open('./tests/data/expense_receipts/v3/receipt.json'))
parsed_receipt = Response.format_response(response, "receipt",
dummy_file_input)