def testColumnStackExecution(self): a_data = np.array((1, 2, 3)) b_data = np.array((2, 3, 4)) a = tensor(a_data, chunk_size=1) b = tensor(b_data, chunk_size=2) c = column_stack((a, b)) res = self.executor.execute_tensor(c, concat=True)[0] expected = np.column_stack((a_data, b_data)) np.testing.assert_equal(res, expected) a_data = np.random.rand(4, 2, 3) b_data = np.random.rand(4, 2, 3) a = tensor(a_data, chunk_size=1) b = tensor(b_data, chunk_size=2) c = column_stack((a, b)) res = self.executor.execute_tensor(c, concat=True)[0] expected = np.column_stack((a_data, b_data)) np.testing.assert_equal(res, expected)
def test_column_stack_execution(setup): a_data = np.array((1, 2, 3)) b_data = np.array((2, 3, 4)) a = tensor(a_data, chunk_size=1) b = tensor(b_data, chunk_size=2) c = column_stack((a, b)) res = c.execute().fetch() expected = np.column_stack((a_data, b_data)) np.testing.assert_equal(res, expected) a_data = np.random.rand(4, 2, 3) b_data = np.random.rand(4, 2, 3) a = tensor(a_data, chunk_size=1) b = tensor(b_data, chunk_size=2) c = column_stack((a, b)) res = c.execute().fetch() expected = np.column_stack((a_data, b_data)) np.testing.assert_equal(res, expected)