自动化面试题，看一下

Question

这是遇到自动化面试题，因为电脑是新电脑，什么都没安装，麻烦做一下 截个图

Answer 1

下面是使用Python、Selenium和pytest/unittest对百度搜索功能进行自动化测试的代码：

import pytest
import unittest
from selenium import webdriver
import time
import os

class BaiduSearch(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(10)
        self.base_url = "https://www.baidu.com"

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

    @pytest.mark.parametrize("search_input", ["python", "selenium", "pytest"])
    def test_baidu_search(self, search_input):
        driver = self.driver
        driver.get(self.base_url)
        driver.find_element_by_id("kw").clear()
        driver.find_element_by_id("kw").send_keys(search_input)
        driver.find_element_by_id("su").click()
        time.sleep(2)

        # 截图并保存
        screenshot_dir = os.path.join(os.getcwd(), "screenshots")
        if not os.path.exists(screenshot_dir):
            os.makedirs(screenshot_dir)

        screenshot_path = os.path.join(screenshot_dir, f"{search_input}.png")
        driver.save_screenshot(screenshot_path)

        # 断言
        assert search_input in driver.title

if __name__ == '__main__':
    unittest.main(verbosity=2, testRunner=pytest.runner.pytest_runner)

1.