Issue #19421: add an unit test for warnings.warn() during finalization

vstinner · vstinner · commit d1b48998e5b4 · 2013-10-28T19:16:21.000+01:00
diff --git a/Lib/test/test_warnings.py b/Lib/test/test_warnings.py
@@ -788,6 +788,25 @@ def test_issue_8766(self):
                 env=env)
             self.assertEqual(retcode, 0)
 
+class FinalizationTest(unittest.TestCase):
+    def test_finalization(self):
+        # Issue #19421: warnings.warn() should not crash
+        # during Python finalization
+        code = """
+import warnings
+warn = warnings.warn
+
+class A:
+    def __del__(self):
+        warn("test")
+
+a=A()
+        """
+        rc, out, err = assert_python_ok("-c", code)
+        # note: "__main__" filename is not correct, it should be the name
+        # of the script
+        self.assertEqual(err, b'__main__:7: UserWarning: test')
+
 
 def setUpModule():
     py_warnings.onceregistry.clear()
