Pyhon学习:测试calculator类

文件名为：calculator.py

#计算器类

class Count:    def __init__(self,a,b):        self.a=int(a)        self.b=int(b)    #加法    def add(self):        return self.a+self.b

测试文件为:test_cal.py

from calculator import Countclass TestCount:    def test_add(self):        j=Count(2,3)        add=j.add()        assert(5==add),'Inter addtion result error!'        except AssertionError as msg:            print(msg)        else:            print('test pass')#执行测试方法;mytest=TestCount()mytest.test_add()

单元测试：test_cal_1.py

from calculator import Countimport unittestclass TestCount(unittest.TestCase):    def setup(self):        print("test start")    def test_add(self):        j=Count(2,3)        self.assertEqual(j.add(),4)    def tearDown(self):        print("test end")if __name__ =='__main__':    unittest.main()