def testUnion1dExecution(self): rs = np.random.RandomState(0) raw1 = rs.random(10) raw2 = rs.random(9) t1 = tensor(raw1, chunk_size=3) t2 = tensor(raw2, chunk_size=4) t = union1d(t1, t2, aggregate_size=1) res = self.executor.execute_tensor(t, concat=True)[0] expected = np.union1d(raw1, raw2) np.testing.assert_array_equal(res, expected) t = union1d(t1, t2) res = self.executor.execute_tensor(t, concat=True)[0] expected = np.union1d(raw1, raw2) np.testing.assert_array_equal(res, expected)
def test_union1d_execution(setup): rs = np.random.RandomState(0) raw1 = rs.random(10) raw2 = rs.random(9) t1 = tensor(raw1, chunk_size=3) t2 = tensor(raw2, chunk_size=4) t = union1d(t1, t2, aggregate_size=1) res = t.execute().fetch() expected = np.union1d(raw1, raw2) np.testing.assert_array_equal(res, expected) t = union1d(t1, t2) res = t.execute().fetch() expected = np.union1d(raw1, raw2) np.testing.assert_array_equal(res, expected)