def test_add_living_space(self): livingspace = LivingSpace('MySpace78') initial_room_count = len(Dojo.rooms()) initial_livingspace_count = len(LivingSpace.rooms()) LivingSpace.add(livingspace) new_room_count = len(Dojo.rooms()) new_livingspace_count = len(LivingSpace.rooms()) self.assertEqual([initial_room_count+1, initial_livingspace_count+1], [new_room_count, new_livingspace_count])
def test_remove_living_space(self): livingspace = LivingSpace('MySpace89') LivingSpace.add(livingspace) initial_room_count = len(Dojo.rooms()) initial_livingspace_count = len(LivingSpace.rooms()) LivingSpace.remove(livingspace) new_room_count = len(Dojo.rooms()) new_livingspace_count = len(LivingSpace.rooms()) self.assertEqual([initial_room_count-1, initial_livingspace_count-1], [new_room_count, new_livingspace_count])