change test file

rerank_esmm.py

@@ -67,18 +67,15 @@ def get_queues(device_id,city_id):
     db = pymysql.connect(host='172.16.40.158', port=4000, user='root',
                          passwd='3SYz54LS9#^9sBvC', db='jerry_test')
     cursor = db.cursor()
-    # sql = "select device_id,city_id from esmm_device_diary_queue " \
-    #       "where device_id = '{}' and city_id = '{}'".format(device_id,city_id)
-    sql = "select device_id,city_id from esmm_device_diary_queue limit 6"
+    sql = "select native_queue, nearby_queue, nation_queue, megacity_queue from esmm_device_diary_queue " \
+          "where device_id = '{}' and city_id = '{}'".format(device_id,city_id)
     cursor.execute(sql)
-    # native_queue, nearby_queue, nation_queue, megacity_queue
-    # if 判断是不是空
-    # result = cursor.fetchone()
-    # print(result)
-    print("a")
-    print(cursor.fetchall())
-    return result
-
+    result = cursor.fetchone()
+    if len(result) > 0:
+        print(cursor.fetchone())
+        return list(cursor.fetchone())
+    else:
+        return []
 
 def tag_boost(cid_list, tag_score):
     pass