57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
# Copyright (C) 2020 by ÿnérant, eichhornchen, nicomarg, charlse
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import unittest
|
|
from random import randint
|
|
from typing import List
|
|
|
|
from squirrelbattle.interfaces import Map, Tile
|
|
from squirrelbattle.mapgeneration import broguelike
|
|
from squirrelbattle.display.texturepack import TexturePack
|
|
|
|
|
|
class TestBroguelike(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.generator = broguelike.Generator()
|
|
self.stom = lambda x : Map.load_from_string("0 0\n" + x)
|
|
self.mtos = lambda x : x.draw_string(TexturePack.ASCII_PACK)
|
|
|
|
def test_dist(self):
|
|
m = self.stom(".. ..\n ... ")
|
|
distance = broguelike.dist(m.tiles, 0, 0, 0, 4)
|
|
self.assertEqual(distance, 6)
|
|
m = self.stom(". .")
|
|
distance = broguelike.dist(m.tiles, 0, 0, 0, 2)
|
|
self.assertEqual(distance, -1)
|
|
|
|
def is_connex(self, grid: List[List[Tile]]) -> bool:
|
|
h, w = len(grid), len(grid[0])
|
|
y, x = randint(0, h - 1), randint(0, w - 1)
|
|
while not (grid[y][x].can_walk()):
|
|
y, x = randint(0, h - 1), randint(0, w - 1)
|
|
queue = Map.neighbourhood(grid, y, x)
|
|
while queue:
|
|
y, x = queue.pop()
|
|
if grid[y][x].can_walk():
|
|
grid[y][x] = Tile.WALL
|
|
queue += Map.neighbourhood(grid, y, x)
|
|
return not any([t.can_walk() for row in grid for t in row])
|
|
|
|
def test_build_doors(self):
|
|
m = self.stom(". .\n. .\n. .\n")
|
|
self.assertFalse(self.generator.build_door(m.tiles, 1, 1, 0, 1, 2))
|
|
|
|
def test_connexity(self) -> None:
|
|
m = self.generator.run()
|
|
self.assertTrue(self.is_connex(m.tiles))
|
|
|
|
def test_loops(self) -> None:
|
|
m = self.stom(3*".. ..\n")
|
|
self.generator.add_loop(m.tiles, 1, 3)
|
|
s = self.mtos(m)
|
|
self.assertEqual(s, ".. ..\n.......\n.. ..")
|
|
self.assertFalse(self.generator.add_loop(m.tiles, 0, 0))
|
|
m = self.stom("...\n. .\n...")
|
|
self.assertFalse(self.generator.add_loop(m.tiles, 1, 1))
|
|
|