@@ -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}