Unittest attempt

1. Inheritance class

import unittest


class TestApi(unittest.TestCase):
    def test_01_login(self):
        print(" Test landing ")

    def test_02_login(self):
        print(" Test registration ")

2. Usage of test suite unittest.main(defaultTest="suite")

import unittest

from test_apis.test_api import TestApi

if __name__ == '__main__':
    suite = unittest.TestSuite()
    suite.addTest(TestApi("test_01_login"))


    #  Add a set of test cases 
    testcases=[TestApi("test_01_login"),TestApi("test_02_login")]
    suite.addTests(testcases)
    unittest.main(defaultTest="suite")

3. Through the loader TestLoader load

import unittest

from test_apis.test_api import TestApi

if __name__ == '__main__':
    suite=unittest.defaultTestLoader.discover("./",pattern="test_*.py")
    unittest.main(defaultTest="suite")

Assert whether the titles are equal ：

import unittest
from selenium import webdriver

class Test_Bai(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:
        driver=webdriver.Chrome()
        driver.get("http://www.baidu.com")
        title=driver.title
        print(title)
        return title

    def test_assert_title(self):
        self.assertEqual(self.setUpClass()," use Baidu Search , You will know ")

if __name__ == '__main__':
    unittest.main