1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
# test_exc.py import unittest def divide(a, b): return a/b class TestIsNumber(unittest.TestCase): def test_1(self): "Divide 4/2" self.assertEqual(divide(4,2), 2) def test_2(self): "Divide 4/0" self.assertRaises(ZeroDivisionError, divide, 4, 0) if __name__ == "__main__": unittest.main()