def test_new_timeseries_error(test_mp):
scen = test_mp.TimeSeries(*test_args, version='new', annotation='testing')
df = {'year': [2010, 2020], 'value': [23.5, 23.6]}
df = pd.DataFrame.from_dict(df)
df['region'] = 'World'
df['variable'] = 'Testing'
pytest.raises(ValueError, scen.add_timeseries, df)
def test_new_timeseries(test_mp):
scen = test_mp.TimeSeries(*test_args, version='new', annotation='testing')
df = {'year': [2010, 2020], 'value': [23.5, 23.6]}
df = pd.DataFrame.from_dict(df)
df['region'] = 'World'
df['variable'] = 'Testing'
df['unit'] = '???'
scen.add_timeseries(df)
scen.commit('importing a testing timeseries')
def test_get_timeseries(test_mp):
scen = test_mp.TimeSeries(*test_args, version=2)
obs = scen.timeseries()
df = {
'region': ['World'],
'variable': ['Testing'],
'unit': ['???'],
'year': [2020],
'value': [23.6]
}
exp = pd.DataFrame.from_dict(df)
npt.assert_array_equal(exp[cols_str], obs[cols_str])
npt.assert_array_almost_equal(exp['value'], obs['value'])
def test_get_timeseries_iamc(test_mp):
scen = test_mp.TimeSeries(*test_args, version=2)
obs = scen.timeseries(iamc=True, regions='World', variables='Testing')
df = {'year': [2010, 2020], 'value': [23.5, 23.6]}
df = pd.DataFrame.from_dict(df)
df['model'] = 'Douglas Adams'
df['scenario'] = 'Hitchhiker'
df['region'] = 'World'
df['variable'] = 'Testing'
df['unit'] = '???'
df = df.pivot_table(index=iamc_idx_cols, columns='year')['value']
df.reset_index(inplace=True)
exp = pd.DataFrame.from_dict(df)
npt.assert_array_equal(exp[iamc_idx_cols], obs[iamc_idx_cols])
npt.assert_array_almost_equal(exp[2010], obs[2010])
