change test file

rerank_esmm.py

@@ -67,13 +67,12 @@ 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 native_queue,nearby_queue,nation_queue,megacity_queue from esmm_device_diary_queue " \
+    sql = "select device_id,city_id from esmm_device_diary_queue " \
           "where device_id = '{}' and city_id = '{}' limit 1".format(device_id,city_id)
     cursor.execute(sql)
+    # native_queue, nearby_queue, nation_queue, megacity_queue
     # if 判断是不是空
-    result = list(cursor.fetchall())
-    print(result)
-    result = list(cursor.fetchone())
+    result = cursor.fetchone()
     print(result)
     return result