diff --git a/tests/pandas_test.py b/tests/pandas_test.py new file mode 100644 index 00000000..610f980f --- /dev/null +++ b/tests/pandas_test.py @@ -0,0 +1,18 @@ +import numpy as np +import pandas as pd +import pyerrors as pe + +def test_df_export_import(tmp_path): + for gz in [True, False]: + my_df = pd.DataFrame([{"int": 1, + "float": -0.01, + "Obs1": pe.pseudo_Obs(87, 21, "test_ensemble"), + "Obs2": pe.pseudo_Obs(-87, 21, "test_ensemble2")}]) + + pe.input.pandas.dump_df(my_df, (tmp_path / 'df_output').as_posix(), gz=gz) + reconstructed_df = pe.input.pandas.load_df((tmp_path / 'df_output').as_posix(), gz=gz) + assert np.all(my_df == reconstructed_df) + + pe.input.pandas.load_df((tmp_path / 'df_output.csv').as_posix(), gz=gz) + +