编写一个简单的单元测试用例

开发一个简单的计算器，用于计算两个数的加减乘除，示例：

class Calculator():
    '''实现简单的加减乘除'''
    def __init__(self,a,b):
        self.a = int(a)
        self.b = int(b)

    def add(self):
        return self.a + self.b

    def sub(self):
        return self.a - self.b

    def mul(self):
        return self.a * self.b

    def div(self):
        return self.a / self.b

使用unittest单元测试框架对Calculator累的方法进行测试。

from module import Calculator
import unittest

class ModuleTest(unittest.TestCase):

    def setUp(self):
        self.cal = Calculator(8,4)

    def tearDown(self):
        pass

    def test_add(self):
        result = self.cal.add()
        self.assertEqual(result,12)

    def test_sub(self):
        result = self.cal.sub()
        self.assertEqual(result,4)

    def test_mul(self):
        result = self.cal.mul()
        self.assertEqual(result,32)

    def test_div(self):
        result = self.cal.div()
        self.assertEqual(result,2)

if __name__ == "__main__":
    #构造测试集
    suite = unittest.TestSuite()
    suite.addTest(ModuleTest("test_add"))
    suite.addTest(ModuleTest("test_sub"))
    suite.addTest(ModuleTest("test_mul"))
    suite.addTest(ModuleTest("test_div"))
    #执行测试
    runer = unittest.TextTestRunner()
    runer.run(suite)