finish the maze generator

tests/test_MazeGenerator.py

@@ -1,11 +1,18 @@
 import numpy
-from amaz_lib.MazeGenerator import Kruskal
+from amaz_lib.MazeGenerator import DepthFirstSearch
 
 
-def test_kruskal_output_shape() -> None:
-    generator = Kruskal()
-    maze = numpy.array([])
-    for output in generator.generator(10, 10):
-        maze = output
+class TestMazeGenerator:
 
-    assert maze.shape == (10, 10)
+    # def test_kruskal_output_shape() -> None:
+    #     generator = Kruskal()
+    #     maze = numpy.array([])
+    #     for output in generator.generator(10, 10):
+    #         maze = output
+
+    #     assert maze.shape == (10, 10)
+
+    def test_generator(self):
+        maze = numpy.array([])
+        maze = DepthFirstSearch.generator(10, 10)
+        assert maze.shape == (10, 10)