def test_lc_predict(calibrator, X):
    calibrator.fit(X, [3, 5])
    assert np.array_equal(calibrator.predict(X), [0, 1])
def test_lc_predict(calibrator, X):
    calibrator.fit(X, [3, 5])
    assert np.array_equal(calibrator.predict(X), [0, 1])