Save test users to a file.

1 files changed, 3 insertions, 3 deletions

diff --git a/monkey/patterns.py b/monkey/patterns.py
index ae958a0..48f9163 100644
--- a/monkey/patterns.py
+++ b/monkey/patterns.py
@@ -336,7 +336,6 @@ def get_patterns(tree):
             for selected in combinations(nodes, 2):
                 pat = pattern2(clause, selected)
                 if pat:
-                    print(pat)
                     yield pat, selected
                 #pat = connect(*selected)
                 #if pat:
@@ -349,7 +348,6 @@ def get_patterns(tree):
             if len(nodes) == 1: 
                 pat = pattern2(clause, nodes)
                 if pat:
-                    print(pat)
                     yield pat, nodes
                 #pat = pattern(tree, var)
                 #if pat:
@@ -446,7 +444,9 @@ if __name__ == '__main__':
     split = int(len(users)*0.7)
     learn_users = set(users[:split])
     test_users = set(users[split:])
-    print(test_users)
+    fusers = open('data/users-test-{}.txt'.format(pid), 'wt')
+    for tu in test_users:
+        fusers.write('{}\n'.format(tu))
 
     for code, data in submissions.items():
         if len(code) > 1000 or prolog_parse(code) is None: