def setUp(self):
self.office = Office('Office')
self.male_room = Living('MaleRoom', 'MALE')
self.female_room = Living('FemaleRoom', 'FEMALE')
self.chidi = Staff('Chidi', 'm')
self.seyi = Staff('Seyi', 'M')
self.lady_seyi = Staff('Seyi', 'f')
self.lade = Fellow('Lade', 'M')
self.fisayo = Fellow('Fisayo', 'M')
self.kate = Fellow('Kate', 'F')
self.sayo = Fellow('Sayo', 'F')
self.office.add_member(self.chidi)
self.office.add_member(self.seyi)
self.office.add_member(self.lady_seyi)
self.male_room.add_member(self.lade)
self.male_room.add_member(self.fisayo)
self.female_room.add_member(self.sayo)
self.female_room.add_member(self.kate)
class TestAmityModels(unittest.TestCase):
def setUp(self):
self.office = Office('Office')
self.male_room = Living('MaleRoom', 'MALE')
self.female_room = Living('FemaleRoom', 'FEMALE')
self.chidi = Staff('Chidi', 'm')
self.seyi = Staff('Seyi', 'M')
self.lady_seyi = Staff('Seyi', 'f')
self.lade = Fellow('Lade', 'M')
self.fisayo = Fellow('Fisayo', 'M')
self.kate = Fellow('Kate', 'F')
self.sayo = Fellow('Sayo', 'F')
self.office.add_member(self.chidi)
self.office.add_member(self.seyi)
self.office.add_member(self.lady_seyi)
self.male_room.add_member(self.lade)
self.male_room.add_member(self.fisayo)
self.female_room.add_member(self.sayo)
self.female_room.add_member(self.kate)
def test_office_instance(self):
self.assertIsInstance(
self.office, Office,
msg="office should be an instance of 'Office' class"
)
def test_office_object_type(self):
self.assertTrue(
(type(self.office) is Office),
msg="The object should be a type of 'Office'"
)
def test_living_instance(self):
self.assertIsInstance(
self.male_room, Living,
msg="living should be an instance of 'Living' class"
)
def test_living_object_type(self):
self.assertTrue(
(type(self.female_room) is Living),
msg="The object should be a type of 'Living space'"
)
def test_staff_instance(self):
self.assertIsInstance(
self.chidi, Staff,
msg="staff should be an instance of 'Staff' class"
)
def test_staff_object_type(self):
self.assertTrue(
(type(self.chidi) is Staff),
msg="The object should be a type of 'Staff'"
)
def test_fellow_instance(self):
self.assertIsInstance(
self.lade, Fellow,
msg="fellow should be an instance of 'Fellow' class"
)
def test_fellow_object_type(self):
self.assertTrue(
(type(self.lade) is Fellow),
msg="The object should be a type of 'Fellow'"
)
def test_can_return_current_size_of_office(self):
self.assertEquals(self.office.current_size(
), 3, msg="The function should return the current number of members in that office")
def test_can_return_current_size_of_living(self):
self.assertEquals(self.male_room.current_size(
), 2, msg="The function should return the current number of members in that living space")
self.assertEquals(self.female_room.current_size(
), 2, msg="The function should return the current number of members in that living space")
