get count

word_vector/tests.py

@@ -18,13 +18,15 @@ if __name__ == "__main__":
         id1_count = 0
         id2_count = 0
         for i in data:
+            print(count)
             count += 1
             tmp = i.split("|")
             device_id = tmp[0]
             ids = tmp[1].rstrip("\n").split(",")
             if id1[0] in ids and id1[1] in ids:
                 id1_count += 1
-            if id2[0] in ids and id2[i] in ids:
+            if id2[0] in ids and id2[1] in ids:
                 id2_count += 1
 
+    print("done")
     print(count, id1_count, id2_count)