diff --git a/squirrelbattle/mapgeneration/randomwalk.py b/squirrelbattle/mapgeneration/randomwalk.py index 32ec0d6..3acb58a 100644 --- a/squirrelbattle/mapgeneration/randomwalk.py +++ b/squirrelbattle/mapgeneration/randomwalk.py @@ -111,5 +111,13 @@ class Generator: result.tiles[y][x] = Tile.FLOOR elif c > 0: result.tiles[y][x] = Tile.WALL + for x in range(width): + for y in [0, height-1]: + if grid[y][x] = Tile.FLOOR: + grid[y][x] = Tile.WALL + for y in range(height): + for y in [0, width-1]: + if grid[y][x] = Tile.FLOOR: + grid[y][x] = Tile.WALL return result