diff --git a/tests/test_companies.py b/tests/test_companies.py index c5d23a7..3acea3e 100644 --- a/tests/test_companies.py +++ b/tests/test_companies.py @@ -1,7 +1,7 @@ import unittest from coffer.units import ( - ACCOUNT_COFFER, ACCOUNT_ORBIT, ACCOUNT_SIGIL, agent_account, company_account, + ACCOUNT_COFFER, ACCOUNT_LEORA, ACCOUNT_NEW_INTERNET, ACCOUNT_ORBIT, ACCOUNT_SIGIL, agent_account, company_account, ) @@ -12,7 +12,8 @@ class CompanyAccountsTest(unittest.TestCase): self.assertNotEqual(ACCOUNT_ORBIT, agent_account("orbit")) def test_company_ids_unique(self): - self.assertEqual(len({ACCOUNT_COFFER, ACCOUNT_SIGIL, ACCOUNT_ORBIT}), 3) + ids = [ACCOUNT_COFFER, ACCOUNT_NEW_INTERNET, ACCOUNT_LEORA, ACCOUNT_SIGIL, ACCOUNT_ORBIT] + self.assertEqual(len(set(ids)), 5) if __name__ == "__main__":