1 files changed, 38 insertions, 0 deletions
diff --git a/pipeline/combine_status_test.py b/pipeline/combine_status_test.py
new file mode 100755
index 0000000..4606587
--- /dev/null
+++ b/pipeline/combine_status_test.py
@@ -0,0 +1,38 @@
+#!/usr/bin/python -S
+"""
+combine_status_test.py: Tests for combine_status.py
+"""
+
+import csv
+import cStringIO
+import unittest
+
+import combine_status  # module under test
+
+
+# TODO: Make these test more the header row.  They rely heavily on the file
+# system!
+
+class CombineStatusTest(unittest.TestCase):
+
+  def testCombineDistTaskStatus(self):
+    stdin = cStringIO.StringIO('')
+    out = cStringIO.StringIO()
+    c_out = csv.writer(out)
+
+    combine_status.CombineDistTaskStatus(stdin, c_out, {})
+    actual = out.getvalue()
+    self.assert_(actual.startswith('job_id,params_file,'), actual)
+
+  def testCombineAssocTaskStatus(self):
+    stdin = cStringIO.StringIO('')
+    out = cStringIO.StringIO()
+    c_out = csv.writer(out)
+
+    combine_status.CombineAssocTaskStatus(stdin, c_out)
+    actual = out.getvalue()
+    self.assert_(actual.startswith('job_id,metric,'), actual)
+
+
+if __name__ == '__main__':
+  unittest.main()