def test_find_cut_no_cut(): y = np.array([0, 0, 0, 0]) k = find_cut(y, 0, len(y)) assert_equal(-1, k)
def test_find_cut(): y = np.array([0, 0, 0, 0, 1, 0, 1, 1]) k = find_cut(y, 0, len(y)) assert_equal(4, k)