diff options
Diffstat (limited to 'migrate')
| -rwxr-xr-x | migrate/tests/fixture/warnings.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/migrate/tests/fixture/warnings.py b/migrate/tests/fixture/warnings.py index 2f54e28..2493708 100755 --- a/migrate/tests/fixture/warnings.py +++ b/migrate/tests/fixture/warnings.py @@ -47,7 +47,10 @@ class catch_warnings(object): """
self._record = record
- self._module = sys.modules['warnings'] if module is None else module
+ if module is None:
+ self._module = sys.modules['warnings']
+ else:
+ self._module = module
self._entered = False
def __repr__(self):
|