change method of checking python version

2026-05-14 20:08:23 +08:00 · 2017-06-29 15:39:14 +08:00
parent 6854607cf0
commit d065864e4b
2 changed files with 4 additions and 7 deletions
--- a/ate/utils.py
+++ b/ate/utils.py
@@ -9,9 +9,10 @@ import yaml
 from ate.exception import ParamsError

 try:
-    assert bytes is str
+    string_type = basestring
    PYTHON_VERSION = 2
-except AssertionError:
+except NameError:
+    string_type = str
    PYTHON_VERSION = 3