def test_add_room_no_branch(self):
builder = LevelBuilder()
room = Room([(1, 0), (0, 0), (0, 1)])
room.neighbours = {}
builder.add_room(room)
self.assertEquals(len(builder.rooms), 1)
added_room = builder.rooms.values()[0]
self.assertEquals(added_room.neighbours, {})
def test_select_branch_room(self):
builder = LevelBuilder()
r1, r2, r3 = object(), object(), object()
builder.rooms = {1: r1, 11: r2, 111: r3}
selected = set()
for _ in xrange(32):
selected.add(builder.select_branch_room())
self.assertEquals(selected, set([r1, r2, r3]))
def test_new_room_verts_tri_on_final_wall(self):
builder = LevelBuilder()
room = Listener()
room.verts = [(1, 11), (2, 22), (3, 33), (4, 44)]
wall = 3
verts = builder.new_room_verts(room, wall, 3)
self.assertEquals(len(verts), 3)
self.assertEquals(verts[0], (1, 11))
self.assertEquals(verts[1], (4, 44))
def test_new_room_verts(self):
builder = LevelBuilder()
room = Listener()
room.verts = [(1, 11), (2, 22), (3, 33), (4, 44)]
wall = 1
for num_verts in xrange(3, 5):
verts = builder.new_room_verts(room, wall, num_verts)
self.assertEquals(len(verts), num_verts)
self.assertEquals(verts[0], (3, 33))
self.assertEquals(verts[1], (2, 22))
def test_select_branch_wall(self):
builder = LevelBuilder()
room = Listener()
room.verts = range(5)
room.neighbours = {0: object(), 2:object(), 3:object()}
walls = set()
for _ in xrange(32):
walls.add(builder.select_branch_wall(room))
self.assertTrue(0 not in walls)
self.assertTrue(2 not in walls)
self.assertTrue(3 not in walls)
def test_add_room(self):
builder = LevelBuilder()
branch_room = Room([(1, 0), (0, 0), (0, 1)])
branch_wall = 5
room = Room([(1, 0), (0, 0), (0, 1)])
room.neighbours = {}
builder.add_room(room, branch_room, branch_wall)
self.assertEquals(len(builder.rooms), 1)
added_room = builder.rooms.values()[0]
self.assertEquals(added_room.neighbours, {0: branch_room})
self.assertEquals(branch_room.neighbours, {branch_wall: added_room})
def test_add_room_works_on_branch_room_and_wall_zero(self):
builder = LevelBuilder()
branch_room = Room([(1, 0), (0, 0), (0, 1)])
branch_room.id = 0
branch_wall = 0
new_room = Room([(1, 0), (0, 0), (0, 1)])
new_room.neighbours = {}
builder.add_room(new_room, branch_room, 0)
self.assertEquals(len(builder.rooms), 1)
added_room = builder.rooms.values()[0]
self.assertEquals(added_room.neighbours, {0: branch_room})
self.assertEquals(branch_room.neighbours, {branch_wall: added_room})
def test_add_room_unions_geometry(self):
builder = LevelBuilder()
room = Room([(0, 0), (0, 1), (1, 0)])
builder.add_room(room)
self.assertEquals(len(builder.geometry.exterior.coords), 4)
self.assertEqual(
list(builder.geometry.exterior.coords),
[(0, 0), (0, 1), (1, 0), (0, 0)])
room = Room([(1, 1), (1, 0), (0, 1)])
builder.add_room(room)
self.assertEquals(len(builder.geometry.exterior.coords), 5)
self.assertEqual(
list(builder.geometry.exterior.coords),
[(0, 0), (0, 1), (1, 1), (1, 0), (0, 0)])
def init(self, name, version):
clock.set_fps_limit(FPS_LIMIT)
self.fps_display = clock.ClockDisplay()
self.camera = Camera((0, 0), 800)
self.renderer = Renderer(self.camera)
caption = '%s v%s' % (name, version)
self.window = Window(
caption=caption, fullscreen=True, visible=False)
self.window.on_key_press = on_key_press
self.window.push_handlers(Keyboard.keystate)
graphics = load_graphics()
self.world = World()
builder = LevelBuilder()
seed(1)
builder.build(self.world, 75, graphics)
self.world.player = Player()
self.world.player.add_to_space(self.world.space, (0, 200), 0)
self.world.chunks.update(self.world.player.chunks)
def test_new_verts_ok(self):
builder = LevelBuilder()
self.assertEquals(builder.new_verts_ok(None), True)
def test_select_branch_wall_no_free_walls(self):
builder = LevelBuilder()
room = Listener()
room.verts = range(3)
room.neighbours = range(3)
self.assertNone(builder.select_branch_wall(room))
def test_create_initial_room(self):
builder = LevelBuilder()
builder.create_initial_room()
added_room = builder.rooms.values()[0]
self.assertEquals(builder.rooms[added_room.id], added_room)
