test_analyze.py 文件源码

def test_Validate():
    """Test that Validate function works correctly"""

    accuracy = an.validate(testing=True)

    val = mean_squared_error(y, slr.predict(X))

    assert np.allclose(accuracy,val)

    accuracy = an.validate(testing=True, X=X, y=y, metric=mean_squared_error)

    assert np.allclose(accuracy,val)

    accuracy = an.validate(testing=True, metric=[mean_squared_error, r2_score])
    val = [mean_squared_error(y, slr.predict(X)), r2_score(y, slr.predict(X))]

    assert np.allclose(accuracy,val)

    with pytest.raises(ValueError):
        an.validate(X=[1,2,3])