Screenshot Example | pytest-html-reporter

test_selenium.py

from selenium import webdriver
import unittest
from pytest_html_reporter import attach

class TestClass(unittest.TestCase):
    def __init__(self, driver):
        super().__init__(driver)

    def setUp(self):
        global driver
        self.driver = webdriver.Chrome()

    def test_demo(self):
        self.driver.get("http://devopsqa.wordpress.com/")
        assert 5 == 4

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

    def screenshot_on_failure(self):
        for self._testMethodName, error in self._outcome.errors:
            if error:
                attach(data=self.driver.get_screenshot_as_png())


if __name__ == '__main__':
    unittest.main()

we need pytest version downgrade (test 7.1.3 ok)
pytest is continuously updated.
It appears that newer versions of pytest are not properly compatible with pytest-html-reporter.