Fix the test for numpy.ndindex()