fixed errors in unit tests

Author: wegar-2

tests/sklearn_extensions/test_log_standard_scaler.py

@@ -7,9 +7,12 @@
 
 
 @fixture(scope="session")
-def make_log_normal_array() -> np.ndarray:
+def make_log_normal_array() -> pd.DataFrame:
     np.random.seed(32123)
-    return np.random.lognormal(mean=2, sigma=4, size=5).reshape(-1, 1)
+    return pd.DataFrame(
+        data=np.random.lognormal(mean=2, sigma=4, size=5).reshape(-1, 1),
+        columns=["X"]
+    )
 
 
 def test_log_standard_scaler_no_shift(make_log_normal_array):
@@ -42,8 +45,3 @@ def test_log_standard_scaler_with_shift_and_base(make_log_normal_array):
     assert round(float(X_trfmd[0, 0]), 8) == 7.12454284
     assert X_trfmd.shape == (5, 1)
     assert round(float(X_trfmd[-1, 0]), 8) == 7.1258186
-
-
-def test_use_of_log_standard_scaler_in_column_transformer(make_log_normal_array):
-    data: pd.DataFrame = pd.DataFrame(data={"X": make_log_normal_array})
-    lss: LogStandardScaler = LogStandardScaler(log_base=2, shift=20)