def testArraySplitExecution(self):
x = arange(48, chunk_size=3).reshape(2, 3, 8)
ss = array_split(x, 3, axis=2)
res = [self.executor.execute_tensor(i, concat=True)[0] for i in ss]
expected = np.array_split(np.arange(48).reshape(2, 3, 8), 3, axis=2)
self.assertEqual(len(res), len(expected))
[np.testing.assert_equal(r, e) for r, e in zip(res, expected)]
ss = array_split(x, [3, 5, 6, 10], axis=2)
res = [self.executor.execute_tensor(i, concat=True)[0] for i in ss]
expected = np.array_split(np.arange(48).reshape(2, 3, 8), [3, 5, 6, 10], axis=2)
self.assertEqual(len(res), len(expected))
[np.testing.assert_equal(r, e) for r, e in zip(res, expected)]
def test_array_split():
a = arange(8, chunk_size=2)
splits = array_split(a, 3)
assert len(splits) == 3
assert [s.shape[0] for s in splits] == [3, 3, 2]
splits = tile(*splits)
assert splits[0].nsplits == ((2, 1), )
assert splits[1].nsplits == ((1, 2), )
assert splits[2].nsplits == ((2, ), )
a = arange(7, chunk_size=2)
splits = array_split(a, 3)
assert len(splits) == 3
assert [s.shape[0] for s in splits] == [3, 2, 2]
splits = tile(*splits)
assert splits[0].nsplits == ((2, 1), )
assert splits[1].nsplits == ((1, 1), )
assert splits[2].nsplits == ((1, 1), )
def testArraySplit(self):
a = arange(8, chunk_size=2)
splits = array_split(a, 3)
self.assertEqual(len(splits), 3)
self.assertEqual([s.shape[0] for s in splits], [3, 3, 2])
splits[0].tiles()
self.assertEqual(splits[0].nsplits, ((2, 1), ))
self.assertEqual(splits[1].nsplits, ((1, 2), ))
self.assertEqual(splits[2].nsplits, ((2, ), ))
a = arange(7, chunk_size=2)
splits = array_split(a, 3)
self.assertEqual(len(splits), 3)
self.assertEqual([s.shape[0] for s in splits], [3, 2, 2])
splits[0].tiles()
self.assertEqual(splits[0].nsplits, ((2, 1), ))
self.assertEqual(splits[1].nsplits, ((1, 1), ))
self.assertEqual(splits[2].nsplits, ((1, 1), ))