lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-11-04 01:25:46 +01:00
Code Issues Projects Releases Packages Wiki Activity

tests: basic test for pandas DataFrame export and re-import added.

Browse source
This commit is contained in:
Fabian Joswig 2022-06-30 14:27:10 +01:00
commit feab699162
1 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/pandas_test.py Normal file
View file

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