✨ Move test from pimms to njab

src/njab/sklearn/__init__.py

@@ -8,13 +8,13 @@
 import pandas as pd
 import sklearn
 import sklearn.model_selection
-
 from mrmr import mrmr_classif
 
-from .types import Splits, ResultsSplit, Results, AucRocCurve, PrecisionRecallCurve
+from . import scoring
 from .pca import run_pca
 from .preprocessing import StandardScaler
-from . import scoring
+from .types import (AucRocCurve, PrecisionRecallCurve, Results, ResultsSplit,
+                    Splits)
 
 __all__ = [
     'run_model',

test/test_sklearn.py

@@ -0,0 +1,12 @@
+import numpy as np
+import numpy.testing as npt
+import pandas as pd
+import sklearn.preprocessing as preprocessing
+
+from njab.sklearn.preprocessing import StandardScaler
+
+
+def test_StandardScaler():
+    X = pd.DataFrame(np.array([[2, None], [3, 2], [4, 6]]))
+    npt.assert_almost_equal(preprocessing.StandardScaler().fit(X).transform(X),
+                            StandardScaler().fit(X).transform(X).to_numpy())