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)