Python+Selenium+unittest自动化测试参数化

from selenium import webdriver  # 导包
import unittest
import time


class login(unittest.TestCase):
    u'''登录'''
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.get("http://xxxxxxxx")
        self.driver.maximize_window()  # 最大化窗口
        self.driver.implicitly_wait(30)  # 隐式等待

    def login(self, username, pwd):
        u'''  定义一个登录的方法,账号和密码参数化  '''
        time.sleep(2)
        self.driver.find_element(by='id', value='loginName').send_keys(username)
        self.driver.find_element(by='id', value='password').send_keys(pwd)
        self.driver.find_element_by_xpath("//*[@type='submit']").click()
        time.sleep(2)


    def is_login_sucess(self):
        try:
            text = self.driver.title
            print(text)
            return True
        except:
            return False

    def test_01(self):
        u'''登录案例:正确账号,正确密码'''
        self.login("xxxx", "xxxx")  # 调用登录方法
        # 判断结果
        result = self.is_login_sucess()
        self.assertTrue(result)

    def test_02(self):
        u'''登录案例:错误账号,错误密码'''
        self.login("22222", "3333")  # 调用登录方法
        # 判断结果
        result = self.is_login_sucess()
        self.assertTrue(result)

    def tearDown(self):
        self.driver.quit()


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