diff --git a/tests/test_hooks.py b/tests/test_hooks.py index 896fa3a7397416704672e941c5e7360197874f3a..bcb7e3ebe7c19575525a82455c18b9dd652349b7 100644 --- a/tests/test_hooks.py +++ b/tests/test_hooks.py @@ -27,6 +27,16 @@ def test_load_all(): ('payee', 'A de B de la C', 'entity', 'de-la-C-A-de-B'), ('corporation', 'Company A', 'corp_entity', 'Company-A'), ('corporation', 'Company A 99', 'corp_entity', 'Company-A-99'), + ('corporation', 'DX Co.', 'corp_entity', 'DX'), + ('corporation', 'DX Company', 'corp_entity', 'DX'), + ('corporation', 'DX Company Inc.', 'corp_entity', 'DX'), + ('corporation', 'DX Corp', 'corp_entity', 'DX'), + ('corporation', 'DX Corp LLC', 'corp_entity', 'DX'), + ('corporation', 'DX Corporation', 'corp_entity', 'DX'), + ('corporation', 'DX, Inc.', 'corp_entity', 'DX'), + ('corporation', 'DX Incorporated', 'corp_entity', 'DX'), + ('payee', 'Poe Inc', 'entity', 'Inc-Poe'), + ('corporation', 'Silly Van', 'corp_entity', 'Silly-Van'), ]) def test_add_entity(in_key, payee, out_key, expected): data = {in_key: payee}