python unittest模块

 import unittest
 import random

 class Operation(object):

     def __init__(self, num1=0, num2=0):
         if not (0 <= num1 <= 10 and 0 <= num2 <= 10):
             raise OutOfRangeError('number out of range (must be 0~10)')
         if not isinstance(num1, int) or not isinstance(num2, int):
             raise NotIntegerError('non‐integers can not be operate')
         self.operate_num1 = num1
         self.operate_num2 = num2

     def get_result(self):
         pass

 class AddOp(Operation):

     def get_result(self):
         return self.operate_num1 + self.operate_num2

 class MinusOp(Operation):

     def get_result(self):
         return self.operate_num1 - self.operate_num2

 class MultiOp(Operation):

     def get_result(self):
         return self.operate_num1 * self.operate_num2

 class DivideOp(Operation):

     def get_result(self):
         return self.operate_num1 / self.operate_num2

 class OperationFactory(object):

     @staticmethod
     def choose_operation(op):
         if op == '+':
             return AddOp()
         elif op == '-':
             return MinusOp()
         elif op == '*':
             return MultiOp()
         elif op == '/':
             return DivideOp()

 class OutOfRangeError(ValueError):
     pass

 class NotIntegerError(ValueError):
     pass

 class KnownValues(unittest.TestCase):

     def test_add_op(self):
         """测试加法运算是否正确"""
         ope_obj = OperationFactory.choose_operation('+')
         for i in range(0, 11):
             ope_obj.operate_num1 = i
             ope_obj.operate_num2 = random.randint(1, 10)
             sum1 = ope_obj.operate_num1 + ope_obj.operate_num2
             sum2 = ope_obj.get_result()
             self.assertEqual(sum1, sum2)

     def test_out_of_range(self):
         """测试出界"""
         for i in [-1, 11]:
             operate_num1 = i
             self.assertRaises(OutOfRangeError, Operation, operate_num1)

     def test_integer(self):
         """测试浮点数"""
         operate_num1 = 0.5
         self.assertRaises(NotIntegerError, Operation, operate_num1)

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