Skip to content

Commit

Permalink
WIP: Test estimator API
Browse files Browse the repository at this point in the history
This isn't ready yet. It currently fails because we don't provide a
get_params function.
  • Loading branch information
timokau committed May 2, 2020
1 parent c4645cd commit 0377ca7
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions csrank/tests/test_estimators.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
"""Check that our estimators adhere to the scikit-learn interface.
https://scikit-learn.org/stable/developers/develop.html#rolling-your-own-estimator
"""

import pytest
from sklearn.utils.estimator_checks import check_estimator

from csrank.choicefunction import FETALinearChoiceFunction
from csrank.discretechoice import FETALinearDiscreteChoiceFunction
from csrank.objectranking import FETALinearObjectRanker


@pytest.mark.parametrize(
"Estimator",
[
FETALinearChoiceFunction,
FETALinearDiscreteChoiceFunction,
FETALinearObjectRanker,
],
)
def test_all_estimators(Estimator):
return check_estimator(Estimator)

0 comments on commit 0377ca7

Please sign in to comment.