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