def setUp(self): self.r = Room('Anvil', 'office') self.l = Room('Ruby', 'living space') self.f = Person.make_person('Jane', 'fellow', sex='F', has_expr_interest=True) self.s = Person.make_person('Mark', 'staff')
def test_can_raise_exception(self): names = ['DAMIAN RICK', 'CHARLES WILLIAM', 'BRAD WILSON', 'WILSON DAMIAN', 'JACK BAUER', 'JASON STATHAM'] persons = [Person.make_person(name, 'Fellow', has_expr_interest=True) for name in names] def assign_all(room_name): for person in persons: Manager.assign_to_room(person, room_name) # raise Exception(Amity.room_collection) self.assertRaises(OutOfOfficeException, assign_all, 'Room 1') self.assertRaises(OutOfLivingSpaceException, assign_all, 'Room 11')