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