关于python+selenium自动化测试中的数据驱动测试

# coding:utf-8
from selenium import webdriver
from public import Login
import time

class Login_data():
    def __init__(self):
        self.driver=webdriver.Firefox(executable_path='/Users/silence/Downloads/geckodriver')
        self.driver.implicitly_wait(4)
        self.driver.get("网址")

    #admin用户登录
    def test_admin_login(self):
        username='登陆账号'
        password='登录密码'
        Login().user_login(self.driver,username,password)
        time.sleep(5)
        print("已使用管理员账号登录")
        self.driver.get_screenshot_as_file("/Users/silence/Documents/screenshot/数据驱动_001.png")
        print("已截图！")
        self.driver.quit()


    #guest用户登录
    def test_guest_login(self):
        username='guset登录账号'
        password='guest登录密码'
        #username=input("")
        #password=input("")
        Login().user_login(self.driver,username,password)
        time.sleep(5)
        print("已使用游客账号登录")
        self.driver.quit()


Login_data().test_admin_login()
Login_data().test_guest_login()