

def _check_behavior_2d_for_constant(clf):
    X = np.array([[0], [0], [0], [0]])
    y = np.array([[1, 0, 5, 4, 3], [2, 0, 1, 2, 5], [1, 0, 4, 5, 2], [1, 3, 3, 2, 0]])
    est = clone(clf)
    est.fit(X, y)
    y_pred = est.predict(X)
    assert_equal(y.shape, y_pred.shape)

dataset/ETHPy150Open scikit-learn/scikit-learn/sklearn/tests/test_dummy.py _check_behavior_2d_for_constant/original