Fix module name with Python 3.5

This commit is contained in:
Xavier Ordoquy 2015-11-11 17:44:46 +01:00
parent 944a1711ec
commit 795cf4716d

View File

@ -186,7 +186,10 @@ class TestUserSetter(TestCase):
try:
self.request.user
except AttributeError as error:
self.assertEqual(str(error), "'module' object has no attribute 'MISSPELLED_NAME_THAT_DOESNT_EXIST'")
assert str(error) in (
"'module' object has no attribute 'MISSPELLED_NAME_THAT_DOESNT_EXIST'", # Python < 3.5
"module 'rest_framework' has no attribute 'MISSPELLED_NAME_THAT_DOESNT_EXIST'", # Python >= 3.5
)
else:
assert False, 'AttributeError not raised'