请问我这个问题怎么解决?
from selenium import webdriver
from selenium.webdriver.common.by import By
import time
from InputElement import InputHandler
from FindElement import find
import pytest
import csv
class BaseTestCase:
@classmethod
def setup_class(cls):
cls.driver = webdriver.Chrome()
url = 'http://www.baidu.com'
cls.driver.get(url)
cls.driver.maximize_window()
cls.find = find(cls.driver)
cls.InputHandler = InputHandler(cls.driver)
@classmethod
def teardown_method(cls):
cls.driver.back()
class TestDemo2(BaseTestCase):
# test_data = [
# ("kw", "测试1", '//div[@id="head"]//div[@id="u"]//a[text()="百度首页"]', 'su', True),
# ("kw", "测试2", '//div[@id="head"]//div[@id="u"]//a[text()="登录"]', 'su', True)
# ]
# # 从别的文件读出来
def read_from_csv(self, file_path):
self.test_data = []
with open(file_path, 'r', newline='') as csvfile:
reader = csv.reader(csvfile)
for row in reader:
id_value, keys_value, xpath_value, click_id, expected_result_str = row
expected_result = expected_result_str == 'True'
# 正确地将数据添加到 self.test_data
self.test_data.append((id_value, keys_value, xpath_value, click_id, expected_result))
return self.test_data
@pytest.mark.parametrize("id_value, keys_value,xpath_value,click_id, expected_result",self.read_from_csv("D:\桌面\新建 Microsoft Excel 工作表.csv"))
def test_01_demo(self,id_value, keys_value, xpath_value, click_id, expected_result):
self.InputHandler.Input_click_element(id_value, keys_value)
self.find.find_element_click(click_id)
assert expected_result == expected_result
def assert_element_exist(self, xpath, id):
try:
element1 = self.driver.find_element(By.ID, id)
except Exception as e:
assert False
elements = self.driver.find_elements(By.XPATH,
xpath)
assert len(elements) > 0