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
