def setup(self):
"""
Setup the test case
"""
self.level = (LevelBuilder().with_size((20, 20)).with_floor_tile(
self.floor_rock).with_wall_tile(self.wall_ground).build())
self.section = new_section((5, 5), (15, 15), self.level, self.rng)
self.generator = SquareRoomGenerator(self.floor_rock, self.wall_empty,
self.floor_rock, ['crypt'])
class TestSquareRoom:
"""
Tests for SquareRoomGenerator room generator
"""
def __init__(self):
"""
Default constructor
"""
self.level = None
self.section = None
self.generator = None
self.floor_empty = 0
self.floor_rock = 1
self.wall_empty = 10
self.wall_ground = 11
self.rng = random.Random()
def setup(self):
"""
Setup the test case
"""
self.level = (
LevelBuilder().with_size((20, 20)).with_floor_tile(self.floor_rock).with_wall_tile(self.wall_ground).build()
)
self.section = new_section((5, 5), (15, 15), self.level, self.rng)
self.generator = SquareRoomGenerator(self.floor_rock, self.wall_empty, self.floor_rock, ["crypt"])
def test_generate_simple_room(self):
"""
Test that generator can create a simple room
"""
self.generator.generate_room(self.section)
room_found = False
for y_loc in range(20):
for x_loc in range(20):
if wall_tile(self.level, (x_loc, y_loc)) != self.wall_empty:
room_found = True
assert_that(room_found, is_(True))
def test_room_connections_are_placed(self):
"""
Test that generating a square room will place 4 connectors
"""
self.generator.generate_room(self.section)
assert_that(list(room_connections(self.section)), has_length(4))
class TestSquareRoom():
"""
Tests for SquareRoomGenerator room generator
"""
def __init__(self):
"""
Default constructor
"""
self.level = None
self.section = None
self.generator = None
self.floor_empty = 0
self.floor_rock = 1
self.wall_empty = 10
self.wall_ground = 11
self.rng = random.Random()
def setup(self):
"""
Setup the test case
"""
self.level = (LevelBuilder().with_size((20, 20)).with_floor_tile(
self.floor_rock).with_wall_tile(self.wall_ground).build())
self.section = new_section((5, 5), (15, 15), self.level, self.rng)
self.generator = SquareRoomGenerator(self.floor_rock, self.wall_empty,
self.floor_rock, ['crypt'])
def test_generate_simple_room(self):
"""
Test that generator can create a simple room
"""
self.generator.generate_room(self.section)
room_found = False
for y_loc in range(20):
for x_loc in range(20):
if wall_tile(self.level, (x_loc, y_loc)) != self.wall_empty:
room_found = True
assert_that(room_found, is_(True))
def test_room_connections_are_placed(self):
"""
Test that generating a square room will place 4 connectors
"""
self.generator.generate_room(self.section)
assert_that(list(room_connections(self.section)), has_length(4))
def setup(self):
"""
Setup the test case
"""
self.level = Level((20, 20), self.floor_rock, self.wall_ground)
self.section = Section((5, 5), (15, 15), self.level, self.rng)
self.generator = SquareRoomGenerator(self.floor_rock, self.wall_empty, ["crypt"])
def setup(self):
"""
Setup the test case
"""
self.level = (
LevelBuilder().with_size((20, 20)).with_floor_tile(self.floor_rock).with_wall_tile(self.wall_ground).build()
)
self.section = new_section((5, 5), (15, 15), self.level, self.rng)
self.generator = SquareRoomGenerator(self.floor_rock, self.wall_empty, self.floor_rock, ["crypt"])
