diff options
Diffstat (limited to 'Lib/importlib/test/util.py')
| -rw-r--r-- | Lib/importlib/test/util.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/Lib/importlib/test/util.py b/Lib/importlib/test/util.py index 7ba7a97..ef32f7d 100644 --- a/Lib/importlib/test/util.py +++ b/Lib/importlib/test/util.py @@ -124,7 +124,11 @@ class mock_modules: else: sys.modules[fullname] = self.modules[fullname] if fullname in self.module_code: - self.module_code[fullname]() + try: + self.module_code[fullname]() + except Exception: + del sys.modules[fullname] + raise return self.modules[fullname] def __enter__(self): |