change test file

5c91b38f · 张彦钊 · 0c95793d · 5c91b38f
Commit 5c91b38f authored Aug 29, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

rerank_esmm.py rerank_esmm.py +3 -2

No files found.
--- a/rerank_esmm.py
+++ b/rerank_esmm.py
@@ -71,9 +71,10 @@ def get_queues(device_id,city_id):
          "where device_id = '{}' and city_id = '{}'".format(device_id,city_id)
    cursor.execute(sql)
    result = cursor.fetchone()
+    db.close()
    if len(result) > 0:
-        print(cursor.fetchone())
-        return list(cursor.fetchone())
+        print(result)
+        return list(result)
    else:
        return []