commit b798ab06937f8bb24b444a49dd42e11fff15e654
author Irit Katriel <iritkatriel@yahoo.com> Tue Feb 23 17:43:04 2021 +0000
committer GitHub <noreply@github.com> Tue Feb 23 09:43:04 2021 -0800
tree c1f19563b4aa8a307f3763fc70731dc004ed0c02
parent 26f18b8540b49d592af66361f8df1a03953d1768

bpo-43146: fix None-handling in single-arg traceback.print_exception(None) (GH-24629)

(The previous commit fixed print_exception(None, None, None).)

Lib/test/test_traceback.py

diff --git a/Lib/test/test_traceback.py b/Lib/test/test_traceback.py
index 2261ea9..5bd969d 100644
--- a/Lib/test/test_traceback.py
+++ b/Lib/test/test_traceback.py
@@ -235,6 +235,10 @@ def test_format_exception_only_exc(self):
     def test_exception_is_None(self):
         NONE_EXC_STRING = 'NoneType: None\n'
         excfile = StringIO()
+        traceback.print_exception(None, file=excfile)
+        self.assertEqual(excfile.getvalue(), NONE_EXC_STRING)
+
+        excfile = StringIO()
         traceback.print_exception(None, None, None, file=excfile)
         self.assertEqual(excfile.getvalue(), NONE_EXC_STRING)
 
@@ -243,6 +247,7 @@ def test_exception_is_None(self):
         self.assertEqual(excfile.getvalue(), NONE_EXC_STRING)
 
         self.assertEqual(traceback.format_exc(None), NONE_EXC_STRING)
+        self.assertEqual(traceback.format_exception(None), [NONE_EXC_STRING])
         self.assertEqual(
             traceback.format_exception(None, None, None), [NONE_EXC_STRING])
         self.assertEqual(