from selenium import webdriver
class TestBaiDu():
def setup(self):
self.driver=webdriver.Chrome()
self.driver.maximize_window()
self.driver.implicitly_wait(5)
def teardown(self):
self.driver.quit()
def test_baidu(self):
self.driver.get("https://www.baidu.com/")
self.driver.find_element_by_css_selector("area").click()
handle=self.driver.window_handles
self.driver.switch_to.window(handle[-1])
el=self.driver.find_element_by_class_name('t')
el.click()
assert '百度热搜 官方'==el.text
示例代码是使用pytest+selenium写的一个简单的测试用例，该用例主要实现以下操作：
第一步：打开百度
第二步：点击页面的百度图标，
第三步：切换另一窗口
第四步：找到“官方热搜”，并点击
第五步：进行断言
通过示例代码，我们要知道，使用pytest+selenium编写测试用例的话，需要有以下四个关键要素：
1. 导入依赖
from selenium import webdriver #导入依赖
使用这条代码，将我们要使用的依赖导入。在【Selenium】（一）Selenium介绍及安装部署说明中我介绍了selenium的原理，我们需要使用webdriver去驱动不同的浏览器，所以当我们使用selenium编写测试用例时一定要导入webdriver。
2. 创建driver
def setup(self):
self.driver=webdriver.Chrome() #声明一个chrome的webdriver
self.driver.maximize_window() #将浏览器窗口最大化
self.driver.implicitly_wait(5) #隐式等待
示例代码中，我们使用setup()和teardown()分别完成用例执行前操作和用例执行后操作。在测试用例执行前，我们需要根据浏览器的种类的来声明的driver，因此我们在setup()中使用self.driver=webdriver.Chrome()声明一个chrome的驱动。
如上图所示，selenium也支持其他类型的浏览器驱动，大家可以根据自己的需求进行使用。
3. 执行自动化步骤
def test_baidu(self):
self.driver.get("https://www.baidu.com/") #打开URL
self.driver.find_element_by_css_selector("area").click() #定位元素并点击
handle=self.driver.window_handles
self.driver.switch_to.window(handle[-1]) #切换窗口
el=self.driver.find_element_by_class_name('t')
el.click()
这里给大家介绍两个常用的操作：
点击：click()
当定位到元素后，使用click()对元素进行点击。
例1：
find_element_by_id('kw').click()
输入：send_keys()
当定位到需要输入内容的元素后，使用send_keys()可以将我们想要输入的内容传送给元素。
例2：
find_element_by_id('kw').send_keys("selenium")
4. 断言
assert '百度热搜 官方'==el.text
测试用例运行后，一般需要使用断言来判断用例是否和我们的预期结果一直，使用assert可以到达这一目的。