def testVStackExecution(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 = vstack([a, b])
res = self.executor.execute_tensor(c, concat=True)[0]
expected = np.vstack([a_data, b_data])
self.assertTrue(np.array_equal(res, expected))
a_data = np.random.rand(10, 20)
b_data = np.random.rand(5, 20)
a = tensor(a_data, chunk_size=3)
b = tensor(b_data, chunk_size=4)
c = vstack([a, b])
res = self.executor.execute_tensor(c, concat=True)[0]
expected = np.vstack([a_data, b_data])
self.assertTrue(np.array_equal(res, expected))
def test_v_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 = vstack([a, b])
res = c.execute().fetch()
expected = np.vstack([a_data, b_data])
assert np.array_equal(res, expected) is True
a_data = np.random.rand(10, 20)
b_data = np.random.rand(5, 20)
a = tensor(a_data, chunk_size=6)
b = tensor(b_data, chunk_size=8)
c = vstack([a, b])
res = c.execute().fetch()
expected = np.vstack([a_data, b_data])
assert np.array_equal(res, expected) is True