change test file

56915fa9 · 张彦钊 · 3b7bd617 · 56915fa9
Commit 56915fa9 authored 5 years ago by 张彦钊
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 3 deletions

rerank_esmm.py rerank_esmm.py +9 -3

No files found.
--- a/rerank_esmm.py
+++ b/rerank_esmm.py
@@ -192,10 +192,16 @@ def to_data_base():
    cursor = db.cursor()
    sql = "select device_id from esmm_device_diary_queue limit 6"
    cursor.execute(sql)
-    result = cursor.fetchone()
+    result = cursor.fetchall()
-    print(result)
-    print(set(result))
    db.close()
+    if result is not None:
+        print([i[0] for i in result])
+        return [i[0] for i in result]
+    else:
+        return []