我的目的：
执行测试脚本，在遇错失败后，截取失败页面图片并保存。

尝试举例：
try:
driver.find_element_by_id("kwsss").send_keys("selenium")
driver.find_element_by_id("su1").click()
except:
driver.get_screenshot_as_file("/home/fnngj/python/error_png.png")

结果：
失败后确实有了截图也保存了，但是我使用的 HTMLtestrunner 生成的测试报告中，测试结果都是 Pass。

求助：
有没有其他更好的办法，能实现报错后截图，测试结果也能保持 fail 的状态。