root / JFP / examples / test_division.py

import unittest
from division import division

class DivisionTestCase(unittest.TestCase):

     def test_one(self):
         self.assertEquals(division(4, 2) , 2)

     def test_two(self):
         self.assert_(division(4, 0) == 0)

def test_suite():
    suite = unittest.TestSuite()
    suite.addTests(unittest.makeSuite(DivisionTestCase))
    return suite

if __name__ == '__main__':
    unittest.main(defaultTest='test_suite')