Test files support both Python 2 and 3

sys-bio · Sep 6, 2018 · 7525c9c · 7525c9c
1 parent 2c12c8d
commit 7525c9c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/wrappers/Python/roadrunner/testing/testfiles.py b/wrappers/Python/roadrunner/testing/testfiles.py
@@ -1,5 +1,5 @@
 import pkgutil
-import os
+import os, sys
 import os.path
 import glob
 import roadrunner
@@ -34,5 +34,8 @@ def getRoadRunner(resource):
     """
     data = pkgutil.get_data(__name__, resource)
     r = roadrunner.RoadRunner()
-    r.load(data.decode())
+    if sys.version_info[0] < 3:
+        r.load(data)
+    else:
+        r.load(data.decode())
     return r