def test_mapcolvals_bad_res_name_len():
"""Testing MapColVals pipeline stages."""
df = _test_df()
value_map = {1: 'Gold', 2: 'Silver', 3: 'Bronze'}
with pytest.raises(ValueError):
map_stage = MapColVals('Medal', value_map, result_columns=['A', 'B'])
assert isinstance(map_stage, MapColVals)
def test_mapcolvals_with_res_name():
"""Testing MapColVals pipeline stages."""
df = _test_df()
value_map = {1: 'Gold', 2: 'Silver', 3: 'Bronze'}
res_df = MapColVals('Medal', value_map, result_columns='Metal').apply(df)
assert res_df['Metal']['UK'] == 'Gold'
assert res_df['Metal']['USSR'] == 'Bronze'
assert res_df['Metal']['US'] == 'Silver'
def test_mapcolvals_no_drop():
"""Testing MapColVals pipeline stages."""
df = _test_df()
value_map = {1: 'Gold', 2: 'Silver', 3: 'Bronze'}
res_df = MapColVals('Medal', value_map, drop=False).apply(df)
assert res_df['Medal']['UK'] == 1
assert res_df['Medal']['USSR'] == 3
assert res_df['Medal']['US'] == 2
assert res_df['Medal_map']['UK'] == 'Gold'
assert res_df['Medal_map']['USSR'] == 'Bronze'
assert res_df['Medal_map']['US'] == 'Silver'
def test_mapcolvals_with_res_name_no_drop():
"""Testing MapColVals pipeline stages."""
df = _test_df()
value_map = {1: 'Gold', 2: 'Silver', 3: 'Bronze'}
map_stage = MapColVals(
'Medal', value_map, result_columns='Metal', drop=False)
res_df = map_stage(df)
assert res_df['Medal']['UK'] == 1
assert res_df['Medal']['USSR'] == 3
assert res_df['Medal']['US'] == 2
assert res_df['Metal']['UK'] == 'Gold'
assert res_df['Metal']['USSR'] == 'Bronze'
assert res_df['Metal']['US'] == 'Silver'