diff --git a/tests/test_MazeGenerator.py b/tests/test_MazeGenerator.py
new file mode 100644
index 0000000..cb0aa6b
--- /dev/null
+++ b/tests/test_MazeGenerator.py
@@ -0,0 +1,11 @@
+import numpy
+from amaz_lib.MazeGenerator import Kruskal
+
+
+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)