print image on test failure

python-visualization · Jun 13, 2020 · abe2062 · abe2062
1 parent 7cc5532
commit abe2062
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/selenium/test_heat_map_selenium.py b/tests/selenium/test_heat_map_selenium.py
@@ -35,4 +35,6 @@ def test_heat_map_with_weights(driver):
     path = os.path.dirname(__file__)
     with open(os.path.join(path, 'test_heat_map_selenium_screenshot.png'), 'rb') as f:
         screenshot_expected = f.read()
+    if not hash(screenshot) == hash(screenshot_expected):
+        print(screenshot)
     assert hash(screenshot) == hash(screenshot_expected)