def test_mean_center(data):
tf = trans.FeatureScaler(method='mean_center')
new_data = tf.fit_transform(data)
assert isinstance(new_data, pd.DataFrame)
# just a few straight comparisons
# following transform() code
means = data.mean(axis=0)
scaled = data.sub(means, axis=1)
assert_frame_equal(new_data, scaled)
def test_range(data):
tf = trans.FeatureScaler(method='range')
new_data = tf.fit_transform(data)
assert isinstance(new_data, pd.DataFrame)
# just a few straight comparisons
# following transform() code
means = data.mean(axis=0)
maxima = data.max(axis=0)
minima = data.min(axis=0)
ranges = maxima - minima
scaled = data.sub(means, axis=1).div(ranges, axis=1)
assert_frame_equal(new_data, scaled)