def test_python_exp_var(rating_true, rating_pred): assert exp_var( rating_true=rating_true, rating_pred=rating_true, col_prediction=DEFAULT_RATING_COL, ) == pytest.approx(1.0, TOL) assert exp_var(rating_true, rating_pred) == pytest.approx(-6.4466, TOL)
def test_python_errors(rating_true, rating_pred): """Test raise errors.""" with pytest.raises(ValueError): rmse(rating_true, rating_true, col_user="******") with pytest.raises(ValueError): mae( rating_pred, rating_pred, col_rating=DEFAULT_PREDICTION_COL, col_user="******", ) with pytest.raises(ValueError): rsquared(rating_true, rating_pred, col_item="not_item") with pytest.raises(ValueError): exp_var( rating_pred, rating_pred, col_rating=DEFAULT_PREDICTION_COL, col_item="not_item", ) with pytest.raises(ValueError): precision_at_k(rating_true, rating_pred, col_rating="not_rating") with pytest.raises(ValueError): recall_at_k(rating_true, rating_pred, col_prediction="not_prediction") with pytest.raises(ValueError): ndcg_at_k(rating_true, rating_true, col_user="******") with pytest.raises(ValueError): map_at_k( rating_pred, rating_pred, col_rating=DEFAULT_PREDICTION_COL, col_user="******", )