encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-03-12 07:55:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use six to reload module

Browse Source
This commit is contained in:
Peter Inglesby 2014-03-06 23:51:02 +00:00
parent 2353878951
commit 29f5ce7aeb
1 changed files with 3 additions and 2 deletions

5
rest_framework/tests/utils.py
View File

@ -1,4 +1,5 @@
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)