fix py2 typeerror bug

mrf345 · Jul 15, 2018 · 6dd799a · 6dd799a
1 parent 9ad02d8
commit 6dd799a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test_fg.py b/test_fg.py
@@ -71,7 +71,7 @@ def test_false_app_gtts(client):
     try:
         gtts(app=app, temporary=200)
     except Exception as e2:
-        assert type(e2) == TypeError
+        assert type(e2) == AssertionError if V.major == 2 else TypeError
     try:
         gtts(app=app, tempdir='/')
     except Exception as e4:
@@ -135,7 +135,7 @@ def test_false_inputs_read(client):
     try:
         eng.read(mouseover=200)
     except Exception as e:
-        assert type(e) == TypeError
+        assert type(e) == AssertionError if V.major == 2 else TypeError
 
 def test_cleanup_before_exit(client):
     """ test cleanup func before exit """