summaryrefslogtreecommitdiff
path: root/monkey
diff options
context:
space:
mode:
Diffstat (limited to 'monkey')
-rw-r--r--monkey/patterns.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/monkey/patterns.py b/monkey/patterns.py
index 8247775..48f9163 100644
--- a/monkey/patterns.py
+++ b/monkey/patterns.py
@@ -444,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: