Private
Public Access
1
0

more tests unit tests on dataprep

This commit is contained in:
Sander Roosendaal
2021-01-18 17:01:59 +01:00
parent 4e8dade3dc
commit 504d3c2d1b
4 changed files with 70 additions and 50 deletions

View File

@@ -543,40 +543,6 @@ def update_c2id_sql(id,c2id):
return 1
def fitnessmetric_to_sql(m,table='powertimefitnessmetric',debug=False,
doclean=False):
# test if nan among values
for key in m.keys():
if str(m[key]) == 'nan':
m[key] = -1
if 'inf' in str(m[key]):
m[key] = -1
if debug:
engine = create_engine(database_url_debug, echo=False)
else:
engine = create_engine(database_url, echo=False)
columns = ', '.join(m.keys())
if use_sqlite:
placeholders = ", ".join(["?"] * len(m))
else:
placeholders = ", ".join(["%s"] * len(m))
query = "INSERT into %s ( %s ) Values (%s)" % (table, columns, placeholders)
query2 = "DELETE FROM %s WHERE PowerFourMin < 0 and PowerOneHour < 0 and PowerTwoK < 0 and user_id = %s " % (table,m['user_id'])
values = tuple(m[key] for key in m.keys())
with engine.connect() as conn, conn.begin():
if doclean:
result2 = conn.execute(query2)
result = conn.execute(query,values)
conn.close()
engine.dispose()
return 1