Use six to reload module

29f5ce7a · Peter Inglesby · 23538789 · 29f5ce7a
Commit 29f5ce7a authored Mar 06, 2014 by Peter Inglesby
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

rest_framework/tests/utils.py
+3 -2

No files found.
--- a/rest_framework/tests/utils.py
+++ b/rest_framework/tests/utils.py
 from contextlib import contextmanager
+from rest_framework.compat import six
 from rest_framework.settings import api_settings


@@ -14,11 +15,11 @@ def temporary_setting(setting, value, module=None):
    setattr(api_settings, setting, value)

    if module is not None:
-        reload(module)
+        six.moves.reload_module(module)

    yield

    setattr(api_settings, setting, original_value)

    if module is not None:
-        reload(module)
+        six.moves.reload_module(module)