def testDStackExecution(self): a_data = np.random.rand(10) b_data = np.random.rand(10) a = tensor(a_data, chunk_size=4) b = tensor(b_data, chunk_size=4) c = dstack([a, b]) res = self.executor.execute_tensor(c, concat=True)[0] expected = np.dstack([a_data, b_data]) self.assertTrue(np.array_equal(res, expected)) a_data = np.random.rand(10, 20) b_data = np.random.rand(10, 20) a = tensor(a_data, chunk_size=3) b = tensor(b_data, chunk_size=4) c = dstack([a, b]) res = self.executor.execute_tensor(c, concat=True)[0] expected = np.dstack([a_data, b_data]) self.assertTrue(np.array_equal(res, expected))
def test_d_stack_execution(setup): a_data = np.random.rand(10) b_data = np.random.rand(10) a = tensor(a_data, chunk_size=8) b = tensor(b_data, chunk_size=8) c = dstack([a, b]) res = c.execute().fetch() expected = np.dstack([a_data, b_data]) assert np.array_equal(res, expected) is True a_data = np.random.rand(10, 20) b_data = np.random.rand(10, 20) a = tensor(a_data, chunk_size=6) b = tensor(b_data, chunk_size=8) c = dstack([a, b]) res = c.execute().fetch() expected = np.dstack([a_data, b_data]) assert np.array_equal(res, expected) is True