class TestOfficeSpaces(unittest.TestCase):
"""Unitest for offices"""
def setUp(self):
self.office = OfficeSpace
self.cedar = OfficeSpace('Cedar', 'OFFICE', 'FELLOW')
self.person = Person('susan', 'adelokiki')
self.people = self.cedar.list_people()
def test_that_a_person_can_be_added_to_office(self):
self.assertEqual(len(self.cedar.people), 0)
self.cedar.add_person(self.person)
self.assertIn(self.person, self.cedar.people)
self.assertEqual(len(self.cedar.people), 1)
def test_that_office_returns_person(self):
self.assertTrue(self.person, self.people)
def setUp(self):
self.office = OfficeSpace
self.cedar = OfficeSpace('Cedar', 'OFFICE', 'FELLOW')
self.person = Person('susan', 'adelokiki')
self.people = self.cedar.list_people()
