当程式变得复杂后,可能需要编写一些测试,并不是孤立检查各个模块,而是要验证模块之间是否能正确地互动。

import unittest

def add(a,b):
    return a+b

def minus(a,b):
    return a-b

def times(a,b):
    return a*b 

def divided(a,b):
    if b != 0:
        return a/b
    else :
        return None
    
class test_math_operation(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(1,1),2)
        self.assertEqual(add(1,-1),0)
        self.assertEqual(add(1,10),11)

    def test_minus(self):
        self.assertEqual(minus(1,1),0)
        self.assertEqual(minus(1,-1),2)
        self.assertEqual(minus(10,1),9)
    
    def test_times(self):
        self.assertEqual(times(1,2),2)
    
    def test_divied(self):
        self.assertEqual(divided(1,1),1)
        self.assertEqual(divided(1,2),0.5)
        self.assertEqual(divided(10,1),10)

if __name__ == '__main__':

    unittest.main()