class TestLivingSpaces(unittest.TestCase):
"""Unitest for offices"""
def setUp(self):
self.room = LivingSpace
self.crucible = LivingSpace('Crucible', 'ROOM', 'MALE')
self.person = Person('stephen', 'sunday')
self.people = self.crucible.list_people()
def test_that_a_person_can_be_added_to_office(self):
self.assertEqual(len(self.crucible.people), 0)
self.crucible.add_person(self.person)
self.assertIn(self.person, self.crucible.people)
self.assertEqual(len(self.crucible.people), 1)
def test_that_office_returns_person(self):
self.assertTrue(self.person, self.people)
def setUp(self):
self.room = LivingSpace
self.crucible = LivingSpace('Crucible', 'ROOM', 'MALE')
self.person = Person('stephen', 'sunday')
self.people = self.crucible.list_people()