calculator.py
class Count:
def __init__(self,a,b):
self.a = a
self.b = b
def add(self):
return self.a + self.b
def sub(self):
return self.a - self.b
test.py
from book.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 test_add2(self):
j = Count(54,49)
self.assertEqual(j.add(),103)
def tearDown(self):
print("test end")
if __name__=="__main__":
#__name__是内置属性,使用__main__就是直接使用。
#main()方法可以将一个单元测试模块变成可以直接运行的测试脚本,执行所有以test命令开头的方法
unittest.main()
test.py
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)
print("test_add执行完毕")
def test_add2(self):
j = Count(54,49)
self.assertEqual(j.add(),103)
print("test_add2执行完毕")
def tearDown(self):
print("test end")
if __name__=="__main__":
#构造测试集
suite = unittest.TestSuite()
suite.addTest(TestCount("test_add2"))
#执行测试
runner = unittest.TextTestRunner()
runner.run(suite)
原文:https://www.cnblogs.com/csj2018/p/10023320.html