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()