def test_standard_scaler_with_std_false_returns_correct_dataframe(
self, numerical: pd.DataFrame):
numerical_scaled = numerical.copy()
numerical_scaled["number_a"] = numerical["number_a"] - 2.5
numerical_scaled["number_b"] = numerical["number_b"] - 6.5
scaler = DFStandardScaler(with_std=False)
result = scaler.fit_transform(numerical)
pd.testing.assert_frame_equal(result, numerical_scaled)
def test_DFStandardScaler_returns_correct_dataframe(numerical):
numerical_scaled = numerical.copy()
numerical_scaled['number_a'] = (numerical['number_a'] -
2.5) / 1.118033988749895
numerical_scaled['number_b'] = (numerical['number_b'] -
6.5) / 1.118033988749895
scaler = DFStandardScaler()
result = scaler.fit_transform(numerical)
pd.testing.assert_frame_equal(result, numerical_scaled)
def test_standard_scaler_returns_correct_dataframe(
self, numerical: pd.DataFrame):
numerical_scaled = numerical.copy()
numerical_scaled["number_a"] = (numerical["number_a"] -
2.5) / 1.118033988749895
numerical_scaled["number_b"] = (numerical["number_b"] -
6.5) / 1.118033988749895
scaler = DFStandardScaler()
result = scaler.fit_transform(numerical)
pd.testing.assert_frame_equal(result, numerical_scaled)