1 files changed, 47 insertions, 0 deletions

diff --git a/runtests.py b/runtests.py
new file mode 100755
index 0000000..88dddf4
--- /dev/null
+++ b/runtests.py
@@ -0,0 +1,47 @@
+#!/usr/bin/python
+
+###############################################################################
+# Description
+###############################################################################
+
+# Unit Tests
+
+
+###############################################################################
+# Imports
+###############################################################################
+
+import trace_reader
+import naive_trace_reader
+import os
+
+###############################################################################
+# Trace files
+###############################################################################
+
+files = [
+'./sample_traces/st-g6-0.bin',
+'./sample_traces/st-g6-1.bin',
+'./sample_traces/st-g6-2.bin',
+'./sample_traces/st-g6-3.bin',
+]
+
+###############################################################################
+# Tests
+###############################################################################
+
+# Does our fancy trace reader get the same number of files as our naive one?
+# (See naive_trace_reader.py for further explanation)
+def test1():
+    stream = trace_reader.trace_reader(files)
+    num_records = len(list(stream))
+    stream = naive_trace_reader.trace_reader(files)
+    naive_num_records = len(list(stream))
+
+    # We need a +1 here because the fancy reader produces a 'meta' record
+    # indicating the number of CPUs
+    if num_records != naive_num_records + 1:
+        return "[FAIL]"
+    return "[SUCCESS]"
+
+print "Test 1: %s" % (test1())