web-automation-demo/Testcase/test_b.py

from Common.Basic import *
from Po.Home.Signin import *


@ddt.ddt
class TestSigninSite(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:
        cls.driver = WebDriver()

    @classmethod
    def tearDownClass(cls) -> None:
        cls.driver.quit()

    def test_screenshot(self):
        self.driver.open('https://www.fanscloud.net/d/index.html')
        self.driver.wait(2)
        self.driver.screenshot_for_report()

    def test_home(self):
        self.driver.open('http://fanscloud.net/')
        self.driver.wait(3)
        self.driver.screenshot_for_report()

if __name__ == '__main__':
    unittest.main()