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)