1#!/usr/bin/python -S 2""" 3combine_results_test.py: Tests for combine_results.py 4""" 5 6import csv 7import cStringIO 8import unittest 9 10import combine_results # module under test 11 12 13# TODO: Make these test more the header row. They rely heavily on the file 14# system! 15 16class CombineResultsTest(unittest.TestCase): 17 18 def testCombineDistResults(self): 19 stdin = cStringIO.StringIO('') 20 out = cStringIO.StringIO() 21 c_out = csv.writer(out) 22 23 combine_results.CombineDistResults(stdin, c_out, 10) 24 actual = out.getvalue() 25 self.assert_(actual.startswith('date'), actual) 26 27 def testCombineAssocResults(self): 28 stdin = cStringIO.StringIO('') 29 out = cStringIO.StringIO() 30 c_out = csv.writer(out) 31 32 combine_results.CombineAssocResults(stdin, c_out, 10) 33 actual = out.getvalue() 34 self.assert_(actual.startswith('dummy'), actual) 35 36 37if __name__ == '__main__': 38 unittest.main() 39