def test_filename(self):
with warnings_state(self.module):
with original_warnings.catch_warnings(record=True, module=self.module) as w:
warning_tests.inner("spam1")
self.assertEqual(os.path.basename(w[-1].filename), "stacklevel.py")
warning_tests.outer("spam2")
self.assertEqual(os.path.basename(w[-1].filename), "stacklevel.py")
def test_filename(self):
with warnings_state(self.module):
with original_warnings.catch_warnings(record=True,
module=self.module) as w:
warning_tests.inner("spam1")
self.assertEqual(os.path.basename(w[-1].filename),
"stacklevel.py")
warning_tests.outer("spam2")
self.assertEqual(os.path.basename(w[-1].filename),
"stacklevel.py")
def test_stacklevel(self):
# Test stacklevel argument
# make sure all messages are different, so the warning won't be skipped
with warnings_state(self.module):
with original_warnings.catch_warnings(record=True,
module=self.module) as w:
warning_tests.inner("spam3", stacklevel=1)
self.assertEqual(os.path.basename(w[-1].filename),
"stacklevel.py")
warning_tests.outer("spam4", stacklevel=1)
self.assertEqual(os.path.basename(w[-1].filename),
"stacklevel.py")
warning_tests.inner("spam5", stacklevel=2)
self.assertEqual(os.path.basename(w[-1].filename),
"__init__.py")
warning_tests.outer("spam6", stacklevel=2)
self.assertEqual(os.path.basename(w[-1].filename),
"stacklevel.py")
warning_tests.outer("spam6.5", stacklevel=3)
self.assertEqual(os.path.basename(w[-1].filename),
"__init__.py")
warning_tests.inner("spam7", stacklevel=9999)
self.assertEqual(os.path.basename(w[-1].filename),
"sys")
def test_stacklevel(self):
with warnings_state(self.module):
with original_warnings.catch_warnings(record=True,
module=self.module) as w:
warning_tests.inner('spam3', stacklevel=1)
self.assertEqual(os.path.basename(w[-1].filename),
'stacklevel.py')
warning_tests.outer('spam4', stacklevel=1)
self.assertEqual(os.path.basename(w[-1].filename),
'stacklevel.py')
warning_tests.inner('spam5', stacklevel=2)
self.assertEqual(os.path.basename(w[-1].filename),
'__init__.py')
warning_tests.outer('spam6', stacklevel=2)
self.assertEqual(os.path.basename(w[-1].filename),
'stacklevel.py')
warning_tests.outer('spam6.5', stacklevel=3)
self.assertEqual(os.path.basename(w[-1].filename),
'__init__.py')
warning_tests.inner('spam7', stacklevel=9999)
self.assertEqual(os.path.basename(w[-1].filename), 'sys')
