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()