Python diagflat示例

示例#1

文件： test_datasource_execute.py 项目： zuodh/mars

    def testDiagflatExecution(self):
        a = diagflat([[1, 2], [3, 4]], chunk_size=1)

        res = self.executor.execute_tensor(a, concat=True)[0]
        expected = np.diagflat([[1, 2], [3, 4]])
        np.testing.assert_equal(res, expected)

        d = tensor([[1, 2], [3, 4]], chunk_size=1)
        a = diagflat(d)

        res = self.executor.execute_tensor(a, concat=True)[0]
        expected = np.diagflat([[1, 2], [3, 4]])
        np.testing.assert_equal(res, expected)

        a = diagflat([1, 2], 1, chunk_size=1)

        res = self.executor.execute_tensor(a, concat=True)[0]
        expected = np.diagflat([1, 2], 1)
        np.testing.assert_equal(res, expected)

        d = tensor([[1, 2]], chunk_size=1)
        a = diagflat(d, 1)

        res = self.executor.execute_tensor(a, concat=True)[0]
        expected = np.diagflat([1, 2], 1)
        np.testing.assert_equal(res, expected)

示例#2

文件： test_datasource_execute.py 项目： fyrestone/mars

def test_diagflat_execution(setup):
    a = diagflat([[1, 2], [3, 4]], chunk_size=1)

    res = a.execute().fetch()
    expected = np.diagflat([[1, 2], [3, 4]])
    np.testing.assert_equal(res, expected)

    d = tensor([[1, 2], [3, 4]], chunk_size=1)
    a = diagflat(d)

    res = a.execute().fetch()
    expected = np.diagflat([[1, 2], [3, 4]])
    np.testing.assert_equal(res, expected)

    a = diagflat([1, 2], 1, chunk_size=1)

    res = a.execute().fetch()
    expected = np.diagflat([1, 2], 1)
    np.testing.assert_equal(res, expected)

    d = tensor([[1, 2]], chunk_size=1)
    a = diagflat(d, 1)

    res = a.execute().fetch()
    expected = np.diagflat([1, 2], 1)
    np.testing.assert_equal(res, expected)