✨ Move test from pimms to njab (#10)

RasmussenLab · Jun 7, 2024 · 90b57d5 · 90b57d5
1 parent da0255b
commit 90b57d5
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/test_sklearn.py b/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())