Test FETADiscreteChoice estimator interface

kiudee · Apr 30, 2020 · dc1ee9f · dc1ee9f
1 parent 2019bff
commit dc1ee9f
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/csrank/tests/test_estimators.py b/csrank/tests/test_estimators.py
@@ -0,0 +1,14 @@
+"""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.discretechoice import FETADiscreteChoiceFunction
+
+
+@pytest.mark.parametrize("Estimator", [FETADiscreteChoiceFunction])
+def test_all_estimators(Estimator):
+    return check_estimator(Estimator)