def test__import_vd(self): fo = { "file": TestToCERO_FileObj._dd + "test__import_vd.VD", "date_col": 3, "val_col": 8 } fo = ToCERO._FileObj(fo) df = fo._import_vd() df.columns.set_names([None], inplace=True) df = df.astype(pd.np.float32) test_df = pd.DataFrame.from_dict( { ("VAR_Act", "-", "FT_COMELC", "ACT", "2015", "PD", "-"): [ 0.740833333333336, 0.740833333333336, 0.8005115537522, 0.829127920241238 ] }, orient="index", dtype=pd.np.float32) test_df.columns = pd.Index([2015, 2016, 2020, 2025]) test_df.sort_index(inplace=True) self.assertTrue(test_df.equals(df)) fo = { "file": TestToCERO_FileObj._dd + "test__import_vd.VD", "date_col": 3, "val_col": 8, "default_year": 2018 } fo = ToCERO._FileObj(fo) df = fo._import_vd() df.columns.set_names([None], inplace=True) df = df.astype(pd.np.float32) df.sort_index(inplace=True) test_df = pd.DataFrame(data=[ [ 0.740833333333336, 0.740833333333336, pd.np.nan, 0.8005115537522, 0.829127920241238 ], [pd.np.nan, pd.np.nan, 1.39891653080538, pd.np.nan, pd.np.nan], [pd.np.nan, pd.np.nan, 19.6047685777802, pd.np.nan, pd.np.nan], [pd.np.nan, pd.np.nan, 31516.8951973493, pd.np.nan, pd.np.nan], ], columns=[2015, 2016, 2018, 2020, 2025], dtype=pd.np.float32) test_df.index = CERO.create_cero_index([ ("VAR_Act", "-", "FT_COMELC", "ACT", "2015", "PD", "-"), ("Cost_Salv", "-", "EN_WinONS-26", "ADE", "2040", "-", "-"), ("Cost_NPV", "-", "EE_StmTurb009", "CQ", "-", "-", "ACT"), ("Reg_irec", "-", "-", "WA", "-", "-", "-"), ]) test_df.sort_index(inplace=True) self.assertTrue(test_df.equals(df))
def test_csv_orientation(self): tc = ToCERO._FileObj( {"file": TestToCERO_FileObj._dd + "test_csv_orientation.csv"}) with self.assertRaises(TypeError): df = tc._import_file() tc = ToCERO._FileObj({ "file": TestToCERO_FileObj._dd + "test_csv_orientation.csv", "orientation": "cols" }) df = tc._import_file() test_df = pd.DataFrame(data=[[1, 2], [3, 4]], columns=[2016, 2017], dtype=pd.np.float32) test_df.index = CERO.create_cero_index(["A", "B"]) test_df.sort_index(inplace=True) self.assertTrue(df.equals(test_df))
def test__import_gdx(self): fo = { "file": TestToCERO_FileObj._dd + "test__import_gdx.gdx", "symbols": { "name": "L_EXPORT", "date_col": 2 } } fo = ToCERO._FileObj(fo) df = fo._import_gdx() with open(TestToCERO_FileObj._dd + "test__import_gdx.pickle", "rb") as f: test_df = pickle.load(f) self.assertTrue(df.equals(test_df))
def test_har_repetitive_sets(self): fo = ToCERO._FileObj({ "file": "test.har", "search_paths": TestToCERO_FileObj._dd, "head_arrs": [{ "name": "ARR7", "default_year": 2018 }] }) cero = fo.import_file_as_cero() df = DataTools.get_test_data(TestToCERO_FileObj._dd + "test_har_repetitive_sets.pickle") self.assertTrue(cero.equals(df))