def test_get_people(self):
office = Office.create("office1")
staff = Staff.create("New Staff")
fellow = Fellow.create("New Fellow")
staff.add_to_room(office)
fellow.add_to_room(office)
self.assertIn(fellow, office.get_people().all())
self.assertIn(staff, office.get_people().all())
def test_create(self):
current_count = Staff.all().count()
Staff.create("New Person")
self.assertEqual(Staff.all().count(), current_count + 1)