def test_split(self):
X = self.make_matrix()
Y = self.make_matrix(cols=2)
[p1, p2, p3, p4] = split(self.sds.from_numpy(X),
self.sds.from_numpy(Y)).compute()
exp1 = X[:2]
exp2 = X[2:]
exp3 = Y[:2]
exp4 = Y[2:]
self.assertTrue(np.allclose(p1, exp1))
self.assertTrue(np.allclose(p2, exp2))
self.assertTrue(np.allclose(p3, exp3))
self.assertTrue(np.allclose(p4, exp4))
def test_split_3(self):
rows = 100
X = self.make_matrix(rows=rows)
Y = self.make_matrix(rows=rows, cols=2)
[p1, p2, p3, p4] = split(self.sds.from_numpy(X),
self.sds.from_numpy(Y)).compute()
exp1 = X[:70]
exp2 = X[70:]
exp3 = Y[:70]
exp4 = Y[70:]
self.assertTrue(np.allclose(p1, exp1))
self.assertTrue(np.allclose(p2, exp2))
self.assertTrue(np.allclose(p3, exp3))
self.assertTrue(np.allclose(p4, exp4))