Draft for tests

Warvito · Oct 16, 2019 · 6e661c2 · 6e661c2
1 parent 45056b0
commit 6e661c2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/neurocombat_sklearn/tests/test_common.py b/neurocombat_sklearn/tests/test_common.py
@@ -2,10 +2,10 @@
 
 from sklearn.utils.estimator_checks import check_estimator
 
-from sklearn_rvm import RVR, RVC
+from neurocombat_sklearn import CombatModel
 
 @pytest.mark.parametrize(
-    "Estimator", [RVR, RVC]
+    "Estimator", [CombatModel]
 )
-def test_all_estimators(Estimator):
+def test_all_transformers(Estimator):
     return check_estimator(Estimator)