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])
