Web自动化【4】——unittest使用 & main()

文件名：calculator.py
类名：Jessi

class Jessi:
    def __init__(self, a, b):
        self.a = a
        self.b = b

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

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

文件名：002_unittest_main.py

mport unittest
from again.base_agin.fun.calculator import Jessi


class TestCal(unittest.TestCase):
    def setUp(self):
        print('start testing')

    def testCase1(self):
        j = Jessi(5, 3)
        # assert j.count() == 8, 'right'
        print('-----------')
        try:
            self.assertEqual(j.minus(), 8, 'error')
        except AssertionError as msg:
            print(msg)

    def testCase2(self):
        j = Jessi(5, 3)
        # assert j.minus() == 3, 'right'
        print('-----------')
        try:
            self.assertEqual(j.minus(), 4, 'error')
        except AssertionError as msg:
            print(msg)

    def tearDown(self):
        print('test end')

if __name__ =='__main__':
	unittest.main()
 # main()使用当前类来搜索包含在该模块中的以‘test’命名开头的测试方法，并自动执行

main()使用当前类来搜索包含在该模块中的以‘test’命名开头的测试方法，并自动执行，所以不管有多少个以‘test’命名的方法，都会执行

那如果只想执行部分以‘test’命名的方法呢？这就需要使用到unittest.TestSuite()对象了。见下篇