def test_default_version(test_mp):
scen = test_mp.Scenario(*can_args)
assert scen.version == 2
def test_years_active(test_mp):
scen = test_mp.Scenario(*msg_multiyear_args)
df = scen.years_active('seattle', 'canning_plant', '2020')
npt.assert_array_equal(df, [2020, 2030])
def test_new_scen(test_mp):
scen = test_mp.Scenario(*can_args, version='new')
assert scen.version == 0
def test_par_filters_unit(test_mp):
scen = test_mp.Scenario(*can_args)
df = scen.par('d', filters={'i': ['seattle']})
obs = df.loc[0, 'unit']
exp = 'km'
assert obs == exp
def test_cat_all(test_mp):
scen = test_mp.Scenario(*msg_args)
df = scen.cat('technology', 'all')
npt.assert_array_equal(df, [
'canning_plant', 'transport_from_seattle', 'transport_from_san-diego'
])
def test_var_level(test_mp):
scen = test_mp.Scenario(*can_args)
df = scen.var('x', filters={'i': ['seattle']})
npt.assert_array_almost_equal(df['lvl'], [50, 300, 0])
def test_var_general_str(test_mp):
scen = test_mp.Scenario(*can_args)
df = scen.var('x', filters={'i': ['seattle']})
npt.assert_array_equal(df['j'], ['new-york', 'chicago', 'topeka'])
def test_var_marginal(test_mp):
scen = test_mp.Scenario(*can_args)
df = scen.var('x', filters={'i': ['seattle']})
npt.assert_array_almost_equal(df['mrg'], [0, 0, 0.036])
def test_idx_name(test_mp):
scen = test_mp.Scenario(*can_args)
df = scen.idx_names('d')
npt.assert_array_equal(df, ['i', 'j'])
def test_init_scalar(test_mp):
scen = test_mp.Scenario(*can_args)
scen2 = scen.clone(keep_sol=False)
scen2.check_out()
scen2.init_scalar('g', 90.0, 'USD/km')
scen2.commit("adding a scalar 'g'")
def test_get_scalar(test_mp):
scen = test_mp.Scenario(*can_args)
obs = scen.scalar('f')
exp = {'unit': 'USD/km', 'value': 90}
assert obs == exp
def test_init_par_35(test_mp):
scen = test_mp.Scenario(*can_args, version='new')
scen.init_set('ii')
scen.init_par('new_par', idx_sets='ii')
