tests: trap exceptions in before_tag

1 files changed, 8 insertions, 5 deletions

diff --git a/tests/environment.py b/tests/environment.py
index 08abbfae..2c30f487 100644
--- a/tests/environment.py
+++ b/tests/environment.py
@@ -30,11 +30,14 @@ def before_all(context):
 
 
 def before_tag(context, tag):
-    # Copy screenshots
-    if tag == 'screenshot':
-        context.save_screenshots = True
-        context.screenshot_dir = "../eog_screenshots"
-        makedirs(context.screenshot_dir)
+    try:
+        # Copy screenshots
+        if 'screenshot' in tag:
+            context.save_screenshots = True
+            context.screenshot_dir = "../eog_screenshots"
+            makedirs(context.screenshot_dir)
+    except Exception as e:
+        print("Error in before_tag: %s" % str(e))
 
 
 def after_step(context, step):