def test_48b():
# tests fix for #48 mapping few->many
exp = IamDataFrame(
pd.DataFrame(
[
['model', 'scen', 'SSD', 'var', 'unit', 1, 6],
['model', 'scen', 'SDN', 'var', 'unit', 1, 6],
['model', 'scen1', 'SSD', 'var', 'unit', 2, 7],
['model', 'scen1', 'SDN', 'var', 'unit', 2, 7],
],
columns=[
'model', 'scenario', 'region', 'variable', 'unit', 2005, 2010
],
)).data
df = IamDataFrame(
pd.DataFrame(
[
['model', 'scen', 'R5MAF', 'var', 'unit', 1, 6],
['model', 'scen1', 'R5MAF', 'var', 'unit', 2, 7],
],
columns=[
'model', 'scenario', 'region', 'variable', 'unit', 2005, 2010
],
))
obs = df.map_regions('iso', region_col='r5_region').data
obs = sort_data(obs[obs.region.isin(['SSD', 'SDN'])], df._LONG_IDX)
pd.testing.assert_frame_equal(obs, exp, check_index_type=False)
def test_48b():
# tests fix for #48 mapping few->many
exp = IamDataFrame(
pd.DataFrame(
[
["model", "scen", "SSD", "var", "unit", 1, 6],
["model", "scen", "SDN", "var", "unit", 1, 6],
["model", "scen1", "SSD", "var", "unit", 2, 7],
["model", "scen1", "SDN", "var", "unit", 2, 7],
],
columns=[
"model", "scenario", "region", "variable", "unit", 2005, 2010
],
)).data
df = IamDataFrame(
pd.DataFrame(
[
["model", "scen", "R5MAF", "var", "unit", 1, 6],
["model", "scen1", "R5MAF", "var", "unit", 2, 7],
],
columns=[
"model", "scenario", "region", "variable", "unit", 2005, 2010
],
))
obs = df.map_regions("iso", region_col="r5_region").data
obs = sort_data(obs[obs.region.isin(["SSD", "SDN"])], df._LONG_IDX)
pd.testing.assert_frame_equal(obs, exp, check_index_type=False)
def _r5_regions_exp(df):
df = df.filter(region='World', keep=False)
df['region'] = 'R5MAF'
return sort_data(df.data, df._LONG_IDX)
def _r5_regions_exp(df):
df = df.filter(region="World", keep=False)
data = df.data
data["region"] = "R5MAF"
return sort_data(data, df._LONG_IDX)
