Print screenshot if test fails

python-visualization · Jun 13, 2020 · 4a2dbc1 · 4a2dbc1
1 parent f68b025
commit 4a2dbc1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/selenium/test_heat_map_selenium.py b/tests/selenium/test_heat_map_selenium.py
@@ -34,4 +34,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()
-    assert hash(screenshot) == hash(screenshot_expected)
+    if hash(screenshot) != hash(screenshot_expected):
+        print(screenshot)
+        assert False, 'screenshot is not as expected'