Python linspace示例

示例#1

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

    def testLinspaceExecution(self):
        a = linspace(2.0, 9.0, num=11, chunk_size=3)

        res = self.executor.execute_tensor(a, concat=True)[0]
        expected = np.linspace(2.0, 9.0, num=11)
        np.testing.assert_allclose(res, expected)

        a = linspace(2.0, 9.0, num=11, endpoint=False, chunk_size=3)

        res = self.executor.execute_tensor(a, concat=True)[0]
        expected = np.linspace(2.0, 9.0, num=11, endpoint=False)
        np.testing.assert_allclose(res, expected)

        a = linspace(2.0, 9.0, num=11, chunk_size=3, dtype=int)

        res = self.executor.execute_tensor(a, concat=True)[0]
        self.assertEqual(res.dtype, np.int_)

示例#2

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

def test_linspace_execution(setup):
    a = linspace(2.0, 9.0, num=11, chunk_size=3)

    res = a.execute().fetch()
    expected = np.linspace(2.0, 9.0, num=11)
    np.testing.assert_allclose(res, expected)

    a = linspace(2.0, 9.0, num=11, endpoint=False, chunk_size=3)

    res = a.execute().fetch()
    expected = np.linspace(2.0, 9.0, num=11, endpoint=False)
    np.testing.assert_allclose(res, expected)

    a = linspace(2.0, 9.0, num=11, chunk_size=3, dtype=int)

    res = a.execute().fetch()
    assert res.dtype == np.int_