def test_read_out_sets(self): gdx_file = os.path.join(os.getcwd(), "tests", "test_database.gdx") create_test_database(gdx_file) df_s = ga.gdx_to_df(gdx_file, "S") self.assertEqual(len(df_s), 10) df_subs = ga.gdx_to_df(gdx_file, "SubS") self.assertEqual(len(df_subs), 10) self.assertEqual(len(df_subs[df_subs["SubS"]]), 5) df_i = ga.gdx_to_df(gdx_file, "I") self.assertEqual(len(df_i), 10) df_subi = ga.gdx_to_df(gdx_file, "SubI") self.assertEqual(len(df_subi), 10) self.assertEqual(len(df_subi[df_subi["SubI"]]), 5) df_subsi = ga.gdx_to_df(gdx_file, "SubSI") self.assertEqual(len(df_subsi), 100) self.assertEqual(len(df_subsi[df_subsi["SubSI"]]), 25) df_subs_empty = ga.gdx_to_df(gdx_file, "SubSEmpty") self.assertEqual(len(df_subs_empty), 10) self.assertEqual(len(df_subs_empty[df_subs_empty["SubSEmpty"]]), 0)
def test_domain_info(self): gdx_file = os.path.join(os.getcwd(), 'tests', 'test_database.gdx') create_test_database(gdx_file) di = DomainInfo(gdx_file) print di self.assertEqual(len(di.symbols), 5)
def test_read_out_parameters(self): gdx_file = os.path.join(os.getcwd(), "tests", "test_database.gdx") create_test_database(gdx_file) df_param_s = ga.gdx_to_df(gdx_file, "Param_S") self.assertEqual(len(df_param_s), 10) self.assertEqual(sum(df_param_s["Param_S"]), 60.0) self.assertEqual(df_param_s.index.names, ["S"]) self.assertEqual(df_param_s.columns, ["Param_S"]) df_param_s_s = ga.gdx_to_df(gdx_file, "Param_S_S") self.assertEqual(len(df_param_s_s), 100) self.assertEqual(sum(df_param_s_s["Param_S_S"]), 0) self.assertEqual(df_param_s_s.index.names, ["S", "SS"]) self.assertEqual(df_param_s_s.columns, ["Param_S_S"]) df_param_s_i = ga.gdx_to_df(gdx_file, "Param_S_I") self.assertEqual(len(df_param_s_i), 100) self.assertEqual(sum(df_param_s_i["Param_S_I"]), 0) self.assertEqual(df_param_s_i.index.names, ["S", "I"]) self.assertEqual(df_param_s_i.columns, ["Param_S_I"])