coffer/tests/test_companies.py

21 lines
697 B
Python

import unittest
from coffer.units import (
ACCOUNT_COFFER, ACCOUNT_LEORA, ACCOUNT_NEW_INTERNET, ACCOUNT_ORBIT, ACCOUNT_SIGIL, agent_account, company_account,
)
class CompanyAccountsTest(unittest.TestCase):
def test_orbit_company_book_is_separate_from_founder(self):
self.assertEqual(ACCOUNT_ORBIT, "company:orbit")
self.assertEqual(company_account("orbit"), ACCOUNT_ORBIT)
self.assertNotEqual(ACCOUNT_ORBIT, agent_account("orbit"))
def test_company_ids_unique(self):
ids = [ACCOUNT_COFFER, ACCOUNT_NEW_INTERNET, ACCOUNT_LEORA, ACCOUNT_SIGIL, ACCOUNT_ORBIT]
self.assertEqual(len(set(ids)), 5)
if __name__ == "__main__":
unittest.main()