lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-03-15 06:40:24 +01:00
Code Issues Projects Releases Packages Wiki Activity

feat: default value for if_exist in to_sql changed to fail, test for

Browse source 
this behavior added.
This commit is contained in:
Fabian Joswig 2022-07-04 17:19:36 +01:00
parent 03d70d3757
commit a6ebcb59bb
2 changed files with 14 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pyerrors/input/pandas.py
View file

@ -7,7 +7,7 @@ from ..correlators import Corr
from .json import create_json_string, import_json_string
def to_sql(df, table_name, db, if_exists="replace", gz=True):
def to_sql(df, table_name, db, if_exists='fail', gz=True):
"""Write DataFrame including Obs or Corr valued columns to sqlite database.
Parameters
@ -30,7 +30,7 @@ def to_sql(df, table_name, db, if_exists="replace", gz=True):
def read_sql_query(sql, db, auto_gamma=False):
"""Execute SQL query on sqlite database and obatin DataFrame including Obs or Corr valued columns.
"""Execute SQL query on sqlite database and obtain DataFrame including Obs or Corr valued columns.
Parameters
----------
@ -113,7 +113,7 @@ def _serialize_df(df, gz=False):
df : pandas.DataFrame
DataFrame to be serilized.
gz: bool
gzip the json string represenation. Default False.
gzip the json string representation. Default False.
"""
out = df.copy()
for column in out:

11
tests/pandas_test.py
View file

@ -1,6 +1,7 @@
import numpy as np
import pandas as pd
import pyerrors as pe
import pytest
def test_df_export_import(tmp_path):
my_dict = {"int": 1,
@ -61,3 +62,13 @@ def test_sql(tmp_path):
for auto_gamma in [False, True]:
re_df = pe.input.pandas.read_sql_query("SELECT * from My_table", my_db, auto_gamma=auto_gamma)
assert np.all(re_df == pe_df)
def test_sql_if_exists_fail(tmp_path):
pe_df = pd.DataFrame([{"Label": 1, "Obs": pe.pseudo_Obs(5 * np.exp(-0.2), 0.01, "test_ensemble", 20)}])
my_db = (tmp_path / "test_db.sqlite").as_posix()
pe.input.pandas.to_sql(pe_df, "My_table", my_db)
with pytest.raises(ValueError):
pe.input.pandas.to_sql(pe_df, "My_table", my_db)
pe.input.pandas.to_sql(pe_df, "My_table", my_db, if_exists='append')
pe.input.pandas.to_sql(pe_df, "My_table", my_db, if_exists='replace')