Skip to content

Commit 90b57d5

Browse files
author
Henry Webel
authored
✨ Move test from pimms to njab (#10)
1 parent da0255b commit 90b57d5

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

test/test_sklearn.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
import numpy as np
2+
import numpy.testing as npt
3+
import pandas as pd
4+
import sklearn.preprocessing as preprocessing
5+
6+
from njab.sklearn.preprocessing import StandardScaler
7+
8+
9+
def test_StandardScaler():
10+
X = pd.DataFrame(np.array([[2, None], [3, 2], [4, 6]]))
11+
npt.assert_almost_equal(preprocessing.StandardScaler().fit(X).transform(X),
12+
StandardScaler().fit(X).transform(X).to_numpy())

0 commit comments

Comments
 (0)