def test_can_generate_and_assign_to_rooms(self):
rooms = alloc.generate_rooms('office')
self.assertEquals(len(rooms), 10)
Amity.add_rooms(rooms)
self.assertIsInstance(rooms[randint(0, 9)], Office)
rooms = alloc.generate_rooms('living space')
self.assertEquals(len(rooms), 10)
Amity.add_rooms(rooms)
self.assertIsInstance(rooms[randint(0, 9)], LivingSpace)
self.assertEqual(Amity.room_count, 20)
# test that file can be parsed
PeopleFileParser.line_to_person(file_path)
person_name = 'ANDREW PHILLIPS'
Manager.assign_to_room(person_name, 'Room 1')
Manager.assign_to_room(person_name, 'Room 10')
person = Amity.find_person(person_name)
self.assertEquals(person.office.name, 'Room 1')
self.assertIn(person, Amity.find_room('Room 1').occupants)
# test that manager can make allocations
Manager.allocate()
self.assertEquals([], Manager.get_list_of_unallocated_people())
def test_can_parse_people_from_file(self):
persons = PeopleFileParser.line_to_person(file_path)
self.assertEquals(len(persons), 5)
self.assertIsInstance(persons[randint(0, 4)], Person)
