Add test for scikit-learn compatibility

sidchaini · Mar 18, 2024 · 07083e3 · 07083e3
1 parent 218a6bb
commit 07083e3
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tests/test_classifier.py b/tests/test_classifier.py
@@ -10,12 +10,11 @@ def test_init():
     assert clf.scale is True
 
 
-# Fix later
-# # Test classifier estimator compatibility with scikit-learn
-# def test_estimator_compatibility():
-#     from sklearn.utils.estimator_checks import check_estimator
+from sklearn.utils.estimator_checks import check_estimator
 
-#     check_estimator(DistanceMetricClassifier())
+
+def test_sklearn_compatibility():
+    check_estimator(DistanceMetricClassifier())
 
 
 # Test fitting the classifier to a dataset