单元测试本身就是通过一段代码验证另一段代码,所以不用单元测试也可以写单元测试
例如
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
写一段程序来验证它
from calculator import Count
class TestCount:
def test_add(self):
try:
j=Count(2,3)
add=j.add()
assert(add ==5),'Integer assition result error!'
except AssertionError as msg:
print(msg)
else:
print('Test pass!')
mytest=TestCount()
mytest.test_add()
接下来使用unittest来验证
from calculator import Count
import unittest
class TestCount(unittest.TestCase):
def setUp(self):
print('test start')
def test_add(self):
j=Count(2,3)
self.assertEqual(j.add(),5)
def tearDown(self):
print('test end')
if __name__ == '__main__':
unittest.main()