unittest实战（二）：用例编写

# coding:utf-8
import unittest
from selenium import webdriver
import time
from ddt import ddt, data, unpack, file_data
import yaml

chrome_driver = r"C:\Program Files (x86)\Python38-32\Lib\site-packages\selenium\webdriver\chrome\chromedriver.exe"

def readfile():
    params = []
    file = open('param.txt', 'r', encoding='utf-8')
    for line in file.readlines():
        params.append(line.strip('\n').split(','))
    print(params)
    return params

class forTesttest(unittest.TestCase):
    def setUp(self) -> None:
        self.driver = webdriver.Chrome(executable_path=chrome_driver)
        self.driver.get('https://www.baidu.com')

    def tearDown(self) -> None:
        time.sleep(5)
        self.driver.quit()

    def test_a(self):
        self.driver.find_element_by_id('kw').send_keys('小龙女')
        self.driver.find_element_by_id('su').click()

    def test_a(self):
        self.driver.find_element_by_id('kw').send_keys('aaaaa')
        self.driver.find_element_by_id('su').click()

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