def test_scale_col_to_target_round(col):
target = 16
result = scl.scale_col_to_target(
col, target, metric='sum', int_result=True)
pdt.assert_series_equal(result, col)
def test_scale_col_to_target_sum(col):
target = 16
expected = col * target / col.sum()
result = scl.scale_col_to_target(col, target, metric='sum')
assert result.sum() == target
pdt.assert_series_equal(result, expected)
def test_scale_col_to_target_mean_median(col, metric):
target = 600
expected = pd.Series([200, 400, 600, 800, 1000])
result = scl.scale_col_to_target(col, target, metric=metric)
assert getattr(result, metric)() == target
pdt.assert_series_equal(result, expected, check_dtype=False)
def test_scale_col_to_target_sum(col):
target = 16
expected = col * target / col.sum()
result = scl.scale_col_to_target(col, target, metric='sum')
assert result.sum() == target
pdt.assert_series_equal(result, expected)
def test_scale_col_to_target_mean_median(col, metric):
target = 600
expected = pd.Series([200, 400, 600, 800, 1000])
result = scl.scale_col_to_target(col, target, metric=metric)
assert getattr(result, metric)() == target
pdt.assert_series_equal(result, expected, check_dtype=False)
def test_scale_col_to_target_round(col):
target = 16
result = scl.scale_col_to_target(col,
target,
metric='sum',
int_result=True)
pdt.assert_series_equal(result, col)
def test_scale_col_to_target_clip(col):
target = 600
clip_low = 450
clip_high = 999
expected = pd.Series([450, 450, 600, 800, 999])
result = scl.scale_col_to_target(
col, target, metric='mean', clip_low=clip_low, clip_high=clip_high)
pdt.assert_series_equal(result, expected, check_dtype=False)
def test_scale_col_to_target_clip(col):
target = 600
clip_low = 450
clip_high = 999
expected = pd.Series([450, 450, 600, 800, 999])
result = scl.scale_col_to_target(col,
target,
metric='mean',
clip_low=clip_low,
clip_high=clip_high)
pdt.assert_series_equal(result, expected, check_dtype=False)
