correct test for dodgy json to take json error messages for diff py v…

…ersions into account.
jonhadfield · Jun 21, 2016 · 4e48f86 · 4e48f86
1 parent 0d54d5b
commit 4e48f86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_users.py b/tests/test_users.py
@@ -44,7 +44,7 @@ def test_get_users_from_json():
 def test_get_users_from_invalid_json():
     with pytest.raises(ValueError) as excinfo:
         Users.from_json(file_loc='{0}/json_input/invalid.json'.format(os.path.dirname(os.path.abspath(__file__))))
-    assert 'No JSON object could be decoded' in str(excinfo.value)
+    assert any((('No JSON object could be decoded' in str(excinfo.value)), ('JSONDecodeError' in str(excinfo.value))))
 
 
 def test_users_repr():