1 files changed, 41 insertions, 0 deletions
diff --git a/binary_search_tool/test/common.py b/binary_search_tool/test/common.py
new file mode 100755
index 00000000..baac9434
--- /dev/null
+++ b/binary_search_tool/test/common.py
@@ -0,0 +1,41 @@
+#!/usr/bin/python2
+"""Common utility functions."""
+
+DEFAULT_OBJECT_NUMBER = 1238
+DEFAULT_BAD_OBJECT_NUMBER = 23
+OBJECTS_FILE = 'objects.txt'
+WORKING_SET_FILE = 'working_set.txt'
+
+
+def ReadWorkingSet():
+  working_set = []
+  f = open(WORKING_SET_FILE, 'r')
+  for l in f:
+    working_set.append(int(l))
+  f.close()
+  return working_set
+
+
+def WriteWorkingSet(working_set):
+  f = open(WORKING_SET_FILE, 'w')
+  for o in working_set:
+    f.write('{0}\n'.format(o))
+  f.close()
+
+
+def ReadObjectsFile():
+  objects_file = []
+  f = open(OBJECTS_FILE, 'r')
+  for l in f:
+    objects_file.append(int(l))
+  f.close()
+  return objects_file
+
+
+def ReadObjectIndex(filename):
+  object_index = []
+  f = open(filename, 'r')
+  for o in f:
+    object_index.append(int(o))
+  f.close()
+  return object_index