fix PR issues

Telefonica · Jan 5, 2022 · 55787de · 55787de
1 parent 394bd9e
commit 55787de
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/toolium/visual_test.py b/toolium/visual_test.py
@@ -414,13 +414,8 @@ def _get_diff_message(message, image_size):
             diff_message = 'Image dimensions do not match'
 
         if diff_message is None:
-            # Check pil engine message
-            m = re.search(r'\(by a distance of (.*)\)', message)
-            if m:
-                diff_message = 'Distance of %0.8f' % (float(m.group(1)) / image_size)
-        if diff_message is None:
-            # Check perceptualdiff engine message
-            m = re.search(r'([0-9]*) pixels are different', message)
+            # Check pil or perceptualdiff engine message
+            m = re.search(r'\(by a distance of (.*)\)', message) or re.search(r'([0-9]*) pixels are different', message)
             if m:
                 diff_message = 'Distance of %0.8f' % (float(m.group(1)) / image_size)
         if diff_message is None: