def test_aggregate_region_missing_all_subregions(): cols = ['model', 'scenario', 'region', 'variable', 'unit', 2005, 2010] data = pd.DataFrame( [['model_a', 'scen_a', 'foo', 'Primary Energy', 'EJ/y', 1, 6], ['model_a', 'scen_a', 'bar', 'Primary Energy', 'EJ/y', 0.75, 5]], columns=cols) df = IamDataFrame(data=data) obs = df.aggregate_region(variable='Primary Energy', region='R5ASIA', subregions=['China', 'Vietnam', 'Japan']) assert len(obs) == 0
def test_aggregate_region_extra_subregion(): cols = ['model', 'scenario', 'region', 'variable', 'unit', 2005, 2010] data = pd.DataFrame( [['model_a', 'scen_a', 'foo', 'Primary Energy', 'EJ/y', 1, 6], ['model_a', 'scen_a', 'bar', 'Primary Energy', 'EJ/y', 0.75, 5]], columns=cols) df = IamDataFrame(data=data) obs = df.aggregate_region(variable='Primary Energy', region='R5ASIA', subregions=['foo', 'bar', 'baz'], components=[], append=False) assert len(obs) == 2