Пример #1
0
    def testZerosExecution(self):
        t = zeros((20, 30), dtype='i8', chunk_size=5)

        res = self.executor.execute_tensor(t, concat=True)
        self.assertTrue(np.array_equal(res[0], np.zeros((20, 30), dtype='i8')))
        self.assertEqual(res[0].dtype, np.int64)

        t2 = zeros_like(t)
        res = self.executor.execute_tensor(t2, concat=True)
        self.assertTrue(np.array_equal(res[0], np.zeros((20, 30), dtype='i8')))
        self.assertEqual(res[0].dtype, np.int64)

        t = zeros((20, 30), dtype='i4', chunk_size=5, sparse=True)
        res = self.executor.execute_tensor(t, concat=True)

        self.assertEqual(res[0].nnz, 0)
Пример #2
0
    def testZerosExecution(self):
        t = zeros((20, 30), dtype='i8', chunk_size=5)

        res = self.executor.execute_tensor(t, concat=True)
        np.testing.assert_array_equal(res[0], np.zeros((20, 30), dtype='i8'))
        self.assertEqual(res[0].dtype, np.int64)

        t2 = zeros_like(t)
        res = self.executor.execute_tensor(t2, concat=True)
        np.testing.assert_array_equal(res[0], np.zeros((20, 30), dtype='i8'))
        self.assertEqual(res[0].dtype, np.int64)

        t = zeros((20, 30), dtype='i4', chunk_size=5, sparse=True)
        res = self.executor.execute_tensor(t, concat=True)

        self.assertEqual(res[0].nnz, 0)

        t = zeros((20, 30), dtype='i8', chunk_size=6, order='F')
        res = self.executor.execute_tensor(t, concat=True)[0]
        expected = np.zeros((20, 30), dtype='i8', order='F')
        self.assertEqual(res.flags['C_CONTIGUOUS'], expected.flags['C_CONTIGUOUS'])
        self.assertEqual(res.flags['F_CONTIGUOUS'], expected.flags['F_CONTIGUOUS'])
Пример #3
0
def test_zeros_execution(setup):
    t = zeros((20, 30), dtype='i8', chunk_size=10)

    res = t.execute().fetch()
    np.testing.assert_array_equal(res, np.zeros((20, 30), dtype='i8'))
    assert res[0].dtype == np.int64

    t2 = zeros_like(t)
    res = t2.execute().fetch()
    np.testing.assert_array_equal(res, np.zeros((20, 30), dtype='i8'))
    assert res[0].dtype == np.int64

    t = zeros((20, 30), dtype='i4', chunk_size=5, sparse=True)
    res = t.execute().fetch()

    assert res[0].nnz == 0

    t = zeros((20, 30), dtype='i8', chunk_size=6, order='F')
    res = t.execute().fetch()
    expected = np.zeros((20, 30), dtype='i8', order='F')
    assert res.flags['C_CONTIGUOUS'] == expected.flags['C_CONTIGUOUS']
    assert res.flags['F_CONTIGUOUS'] == expected.flags['F_CONTIGUOUS']