test/testColsum.py

   1 '''
   2 Created on Dec 2, 2010
   3
   4 @author: sau
   5 '''
   6 import unittest
   7 import os
   8 from erange import colsum
   9
  10
  11 class TestColsum(unittest.TestCase):
  12
  13     testInputFile = "erangeTestColsumFile"
  14
  15
  16     def setUp(self):
  17         colfile = open(self.testInputFile, "w")
  18         colfile.write("line1 30 60.5\n")
  19         colfile.write("line2 90 99\n")
  20         colfile.write("line3 5 80\n")
  21         colfile.write("line4 10 1\n")
  22         colfile.close()
  23
  24
  25     def tearDown(self):
  26         try:
  27             os.remove(self.testInputFile)
  28         except OSError:
  29             print "cdf file does not exist"
  30
  31
  32     def testBinsToCdf(self):
  33         self.assertEquals(0, colsum.colsum(0, self.testInputFile))
  34         self.assertEquals(135, colsum.colsum(1, self.testInputFile))
  35         self.assertEquals(240.5, colsum.colsum(2, self.testInputFile))
  36         self.assertEquals(0, colsum.colsum(3, self.testInputFile))
  37
  38
  39     def testMain(self):
  40         field = 1
  41         argv = ["colsum"]
  42         self.assertRaises(SystemExit, colsum.main, argv)
  43         argv = ["colsum", field, self.testInputFile]
  44         colsum.main(argv)
  45
  46
  47 def suite():
  48     suite = unittest.TestSuite()
  49     suite.addTest(unittest.makeSuite(TestColsum))
  50
  51     return suite
  52
  53
  54 if __name__ == "__main__":
  55     #import sys;sys.argv = ['', 'Test.testName']
  56     unittest.main()