Python indicesの例

コード例 #1

ファイル: test_datasource_execute.py プロジェクト: zuodh/mars

    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])

コード例 #2

ファイル: test_datasource_execute.py プロジェクト: fyrestone/mars

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])