#encoding=utf-8from selenium import webdriverimport timeimport traceback#获取测试数据with open("data.txt") as file_obj: datas = file_obj.readlines()result = []#遍历测试数据for data in datas: try: driver = webdriver.Chrome(executable_path = "e:\\chromedriver") url = "https://www.baidu.com" driver.get(url) driver.find_element_by_id("kw").send_keys(data.strip().split("||")[0].decode("gbk")) driver.find_element_by_id("su").click() time.sleep(3) assert data.strip().split("||")[1].decode("gbk") in driver.page_source except Exception as e: traceback.print_exc() #因为结果是ANSI格式保存的，所以编码为gbk，不至于乱码 result.append(data.strip().split("||")[0] +u"失败\n".encode("gbk")) else: result.append(data.strip().split("||")[0]+u"成功\n".encode("gbk")) time.sleep(2)#测试结果写入文件，result列表的内容需要时字符串 with open("result.txt","a") as file_obj: file_obj.writelines(result)driver.quit()