Coverage for tests/unit/generation/test_corner_first_ndindex.py: 100%

1import numpy as np 

2 

3from maze_dataset.utils import corner_first_ndindex 

4 

5 

6def test_corner_first_ndindex(): 

7 for n in range(1, 11): 

8 a_n = corner_first_ndindex(n) 

9 a_n_plus_1 = corner_first_ndindex(n + 1) 

10 assert np.all(a_n == a_n_plus_1[: n**2])