def test_equal_two_dimensional_split(self):
matrix = np.reshape(np.arange(16), (4, 4))
res = split(matrix, [4, 2], 0, True)
desired = [[[0, 1]], [[2, 3]], [[4, 5]], [[6, 7]], [[8, 9]], [[10,
11]],
[[12, 13]], [[14, 15]]]
compare_results(res, desired)
res = split(matrix, [2, 2], 0, True)
desired = [[[0, 1], [4, 5]], [[2, 3], [6, 7]], [[8, 9], [12, 13]],
[[10, 11], [14, 15]]]
compare_results(res, desired)
def test_equal_split(self):
a = np.arange(10)
res = split(a, 2)
desired = [np.arange(5), np.arange(5, 10)]
compare_results(res, desired)
def test_equal_split(self):
a = np.arange(10)
res = split(a, 2)
desired = [np.arange(5), np.arange(5, 10)]
compare_results(res, desired)
