mirror of
https://github.com/fjosw/pyerrors.git
synced 2025-03-15 14:50:25 +01:00
tests: basic test for pandas DataFrame export and re-import added.
This commit is contained in:
parent
f980229d5c
commit
feab699162
1 changed files with 18 additions and 0 deletions
18
tests/pandas_test.py
Normal file
18
tests/pandas_test.py
Normal 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)
|
||||
|
||||
|
Loading…
Add table
Reference in a new issue