diff options
| -rw-r--r-- | Lib/test/test_warnings.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/Lib/test/test_warnings.py b/Lib/test/test_warnings.py index 11dc294..bc7e398 100644 --- a/Lib/test/test_warnings.py +++ b/Lib/test/test_warnings.py @@ -553,10 +553,13 @@ class _WarningsTests(BaseTest): globals_dict = globals() oldfile = globals_dict['__file__'] try: - with original_warnings.catch_warnings(module=self.module) as w: + with original_warnings.catch_warnings(module=self.module, record=True) as w: self.module.filterwarnings("always", category=UserWarning) globals_dict['__file__'] = None self.module.warn('test', UserWarning) + self.assertEqual(len(w), 1) + self.assertEqual(w[0].category, UserWarning) + self.assertEqual(str(w[0].message), 'test') finally: globals_dict['__file__'] = oldfile |