Raise "VisualException" for visual diff assertion failures

Author: mdmintz

seleniumbase/common/exceptions.py

@@ -4,25 +4,29 @@
     OutOfScopeException => Used by BaseCase methods when setUp() is skipped.
     TextNotVisibleException => Called when expected text fails to appear.
     TimeLimitExceededException => Called when exceeding "--time-limit=SECONDS".
+    VisualException => Called when there's a Visual Diff Assertion Failure.
 """
-from selenium.common.exceptions import WebDriverException
 
 
 class NoSuchFileException(Exception):
     pass
 
 
-class NotUsingChromeException(WebDriverException):
+class NotUsingChromeException(Exception):
     pass
 
 
 class OutOfScopeException(Exception):
     pass
 
 
-class TextNotVisibleException(WebDriverException):
+class TextNotVisibleException(Exception):
     pass
 
 
 class TimeLimitExceededException(Exception):
     pass
+
+
+class VisualException(Exception):
+    pass

seleniumbase/fixtures/base_case.py

@@ -11330,7 +11330,9 @@ def __assert_eq(self, *args, **kwargs):
                 elif line.strip().startswith("*"):
                     minified_exception += line + "\n"
         if minified_exception:
-            raise Exception(minified_exception)
+            from seleniumbase.common.exceptions import VisualException
+
+            raise VisualException(minified_exception)
 
     def __process_visual_baseline_logs(self):
         """Save copies of baseline PNGs in "./latest_logs" during failures.