def testIndicesExecution(self): grid = indices((2, 3), chunk_size=1) res = self.executor.execute_tensor(grid, concat=True)[0] expected = np.indices((2, 3)) np.testing.assert_equal(res, expected) res = self.executor.execute_tensor(grid[0], concat=True)[0] np.testing.assert_equal(res, expected[0]) res = self.executor.execute_tensor(grid[1], concat=True)[0] np.testing.assert_equal(res, expected[1])
def test_indices_execution(setup): grid = indices((2, 3), chunk_size=1) res = grid.execute().fetch() expected = np.indices((2, 3)) np.testing.assert_equal(res, expected) res = grid[0].execute().fetch() np.testing.assert_equal(res, expected[0]) res = grid[1].execute().fetch() np.testing.assert_equal(res, expected[1])