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