Back to previous test

Author: hamidkm9

tests/test_linearboost.py

@@ -1,7 +1,16 @@
-from sklearn.utils.estimator_checks import parametrize_with_checks
+from sklearn.base import is_classifier
+
 from linearboost.linear_boost import LinearBoostClassifier
 
+from ._utils import check_estimator, get_expected_failed_tests
+
 
-@parametrize_with_checks([LinearBoostClassifier()])
-def test_sklearn_compatible_estimator(estimator, check):
-    check(estimator)
+def test_linear_boost_estimator():
+    """
+    Test whether `LinearBoostClassifier` adheres to scikit-learn conventions.
+    """
+    check_estimator(
+        LinearBoostClassifier(),
+        expected_failed_checks=get_expected_failed_tests(LinearBoostClassifier()),
+    )
+    assert is_classifier(LinearBoostClassifier)