finish the generator DFS

This commit is contained in:
Maoake Teriierooiterai
2026-03-24 14:28:10 +01:00
parent 8b4ef7afce
commit 4d151664ab
5 changed files with 44 additions and 46 deletions
+8 -7
View File
@@ -1,18 +1,19 @@
import numpy
from amaz_lib.MazeGenerator import DepthFirstSearch
from amaz_lib.MazeGenerator import Kruskal
class TestMazeGenerator:
# def test_kruskal_output_shape() -> None:
# generator = Kruskal()
# maze = numpy.array([])
# for output in generator.generator(10, 10):
# maze = output
def test_kruskal_output_shape() -> None:
generator = Kruskal().generator(10, 10)
maze = numpy.array([])
for output in generator:
maze = output
# assert maze.shape == (10, 10)
assert maze.shape == (10, 10)
def test_generator(self):
def test_generator(self) -> None:
maze = numpy.array([])
maze = DepthFirstSearch.generator(10, 10)
assert maze.shape == (10, 10)