From fbc039c27422428ed5889adda449f69d65076255 Mon Sep 17 00:00:00 2001
From: David Taylor <david@taylorhq.com>
Date: Tue, 16 May 2017 13:58:05 +0100
Subject: [PATCH] Fix tests so they can actually run locally (I failed)

---
 RIGS/test_functional.py | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

diff --git a/RIGS/test_functional.py b/RIGS/test_functional.py
index 9452e41c..a804ddd5 100644
--- a/RIGS/test_functional.py
+++ b/RIGS/test_functional.py
@@ -26,15 +26,20 @@ browsers = [{"platform": "macOS 10.12",
 
 
 def on_platforms(platforms):
-    if os.environ.get("TRAVIS"):
-        def decorator(base_class):
-            module = sys.modules[base_class.__module__].__dict__
-            for i, platform in enumerate(platforms):
-                d = dict(base_class.__dict__)
-                d['desired_capabilities'] = platform
-                name = "%s_%s" % (base_class.__name__, i + 1)
-                module[name] = type(name, (base_class,), d)
-        return decorator
+    if not os.environ.get("TRAVIS"):
+        platforms = {'local'}
+
+    def decorator(base_class):
+        module = sys.modules[base_class.__module__].__dict__
+        for i, platform in enumerate(platforms):
+            d = dict(base_class.__dict__)
+            d['desired_capabilities'] = platform
+            name = "%s_%s" % (base_class.__name__, i + 1)
+            module[name] = type(name, (base_class,), d)
+    
+    return decorator
+
+
 
 
 def create_browser(test_name, desired_capabilities):