more tests unit tests on dataprep
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user