def test_level_generator_is_created(self):
"""
Test that level generator is created for newly created portal
"""
level = Level(size = (20, 20),
floor_type = self.floor_rock,
wall_type = self.wall_empty)
level_generator = mock(LevelGenerator)
for loc_y in range(8, 12):
for loc_x in range(8, 12):
level.set_location_type((loc_x, loc_y), 'room')
portal_adder = PortalAdder((1, 2),
'room',
level_generator,
self.rng)
portal_adder.add_portal(level)
portals = level.portals
portal = level.portals[0]
assert_that(portal.level_generator, is_(same_instance(level_generator)))
def test_portal_has_icons(self):
"""
Test that portal created by adder has two icons set
One to display and another to be used by opposite end
"""
level = Level(size = (20, 20),
floor_type = self.floor_rock,
wall_type = self.wall_empty)
level_generator = mock(LevelGenerator)
for loc_y in range(8, 12):
for loc_x in range(8, 12):
level.set_location_type((loc_x, loc_y), 'room')
portal_adder = PortalAdder((1, 2),
'room',
level_generator,
self.rng)
portal_adder.add_portal(level)
portals = level.portals
portal = level.portals[0]
assert_that(portal.icon, is_(equal_to(1)))
assert_that(portal.other_end_icon, is_(equal_to(2)))
def test_add_stairs_to_room(self):
"""
Test that stairs can be added to a room
"""
level = Level(size = (20, 20),
floor_type = self.floor_rock,
wall_type = self.wall_empty)
for loc_y in range(8, 12):
for loc_x in range(8, 12):
level.set_location_type((loc_x, loc_y), 'room')
portal_adder = PortalAdder((1, 2),
'room',
mock(),
self.rng)
portal_adder.add_portal(level)
portals = level.portals
assert_that(portals, has_length(1))
portal = level.portals[0]
assert_that(located_in_room(portal), is_(True))
