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)