自动化用例是由程序去执行，因此有时候打印的错误信息并不明确，如果在脚本执行错误的时候能对当前窗口截图保存，那么通过图片就可以非常直观的看出出错的原因。webdriver提供了截图函数get_screenshot_as_file()来截取当前创窗口。
1 from selenium import webdriver 2 from time import sleep 3 driver = webdriver.Firefox() 4 driver.get("http://www.baidu.com") 5 6 driver.find_element_by_id('kw').send_keys('selenium') 7 driver.find_element_by_id('su').click() 8 sleep(2) 9 10 #截取当前窗口，指定截图图片的保存位置 11 driver.get_screenshot_as_file(r"C:UsersfyrDesktop123.jpg") 12 driver.quit()