def test_scale_to_targets_no_segments(df, target_col):
target = [1000]
metric = 'mean'
result = scl.scale_to_targets(df, target_col, target, metric=metric)
pdt.assert_index_equal(result.columns, df.columns)
pdt.assert_series_equal(
result[target_col],
pd.Series(
[181.81818182, 363.63636364, 545.45454545, 727.27272727,
909.09090909, 1090.90909091, 1272.72727273, 1454.54545455,
1636.36363636, 1818.18181818]),
check_dtype=False)
def test_scale_to_targets(df, target_col):
targets = [100, 1000]
filters = [['geo_id == "a"', 'filter_col < 106'], 'geo_id == "b"']
metric = 'sum'
result = scl.scale_to_targets(df, target_col, targets, metric, filters)
pdt.assert_index_equal(result.columns, df.columns)
pdt.assert_series_equal(
result[target_col],
pd.Series(
[11.11111111, 66.66666667, 33.33333333, 133.33333333, 55.55555556,
200, 7, 266.66666667, 9, 333.33333333]),
check_dtype=False)
def test_scale_to_targets_clip_int(df, target_col):
target = [1000]
metric = 'mean'
clip_low = 400
clip_high = 999.99
int_result = True
result = scl.scale_to_targets(
df, target_col, target, metric, clip_low=clip_low, clip_high=clip_high,
int_result=int_result)
pdt.assert_index_equal(result.columns, df.columns)
pdt.assert_series_equal(
result[target_col],
pd.Series([400, 400, 545, 727, 909, 1000, 1000, 1000, 1000, 1000]))
def test_scale_to_targets_no_segments(df, target_col):
target = [1000]
metric = 'mean'
result = scl.scale_to_targets(df, target_col, target, metric=metric)
pdt.assert_index_equal(result.columns, df.columns)
pdt.assert_series_equal(result[target_col],
pd.Series([
181.81818182, 363.63636364, 545.45454545,
727.27272727, 909.09090909, 1090.90909091,
1272.72727273, 1454.54545455, 1636.36363636,
1818.18181818
]),
check_dtype=False)
def test_scale_to_targets(df, target_col):
targets = [100, 1000]
filters = [['geo_id == "a"', 'filter_col < 106'], 'geo_id == "b"']
metric = 'sum'
result = scl.scale_to_targets(df, target_col, targets, metric, filters)
pdt.assert_index_equal(result.columns, df.columns)
pdt.assert_series_equal(result[target_col],
pd.Series([
11.11111111, 66.66666667, 33.33333333,
133.33333333, 55.55555556, 200, 7,
266.66666667, 9, 333.33333333
]),
check_dtype=False)
def test_scale_to_targets_clip_int(df, target_col):
target = [1000]
metric = 'mean'
clip_low = 400
clip_high = 999.99
int_result = True
result = scl.scale_to_targets(df,
target_col,
target,
metric,
clip_low=clip_low,
clip_high=clip_high,
int_result=int_result)
pdt.assert_index_equal(result.columns, df.columns)
pdt.assert_series_equal(
result[target_col],
pd.Series([400, 400, 545, 727, 909, 1000, 1000, 1000, 1000, 1000]))