add test

tensnsorflow/ffm.py

@@ -226,7 +226,8 @@ def get_predict_set(ucity_id,model,ccity_name,manufacturer,channel,level2_ids):
           "u.device_type,u.manufacturer,u.channel,c.top,df.level2_ids,e.device_id,e.cid_id " \
           "from esmm_pre_data e left join user_feature u on e.device_id = u.device_id " \
           "left join cid_type_top c on e.device_id = c.device_id " \
-          "left join diary_feat df on e.cid_id = df.diary_id"
+          "left join diary_feat df on e.cid_id = df.diary_id " \
+          "where e.device_id = 'B7B00128-7081-4CA7-9237-5001AA1A48D1'"
     df = con_sql(db, sql)
     df = df.rename(columns={0: "y", 1: "z", 2: "label", 3: "ucity_id", 4: "clevel1_id", 5: "ccity_name",
                             6: "device_type", 7: "manufacturer", 8: "channel", 9: "top", 10: "level2_ids",