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')