add test

Author: methane

Lib/test/test_traceback.py

@@ -564,12 +564,12 @@ def test_signatures(self):
         self.assertEqual(
             str(inspect.signature(traceback.print_exception)),
             ('(exc, /, value=<implicit>, tb=<implicit>, '
-             'limit=None, file=None, chain=True, **kwargs)'))
+             'limit=None, file=None, chain=True, show_lines=True, **kwargs)'))
 
         self.assertEqual(
             str(inspect.signature(traceback.format_exception)),
             ('(exc, /, value=<implicit>, tb=<implicit>, limit=None, '
-             'chain=True, **kwargs)'))
+             'chain=True, show_lines=True, **kwargs)'))
 
         self.assertEqual(
             str(inspect.signature(traceback.format_exception_only)),
@@ -3340,7 +3340,7 @@ def some_inner(k, v):
 
     def test_custom_format_frame(self):
         class CustomStackSummary(traceback.StackSummary):
-            def format_frame_summary(self, frame_summary, colorize=False):
+            def format_frame_summary(self, frame_summary, **kwargs):
                 return f'{frame_summary.filename}:{frame_summary.lineno}'
 
         def some_inner():
@@ -3365,10 +3365,10 @@ def g():
         tb = g()
 
         class Skip_G(traceback.StackSummary):
-            def format_frame_summary(self, frame_summary, colorize=False):
+            def format_frame_summary(self, frame_summary, **kwargs):
                 if frame_summary.name == 'g':
                     return None
-                return super().format_frame_summary(frame_summary)
+                return super().format_frame_summary(frame_summary, **kwargs)
 
         stack = Skip_G.extract(
             traceback.walk_tb(tb)).format()
@@ -4877,5 +4877,161 @@ def expected(t, m, fn, l, f, E, e, z):
         ]
         self.assertEqual(actual, expected(**colors))
 
+
+class TestShowLines(unittest.TestCase):
+    """Tests for the show_lines parameter in traceback formatting functions."""
+
+    def setUp(self):
+        # Create a simple exception for testing
+        try:
+            x = 1 / 0
+        except ZeroDivisionError as e:
+            self.exc = e
+
+    def test_print_tb_show_lines_true(self):
+        """Test print_tb with show_lines=True (default)"""
+        output = StringIO()
+        traceback.print_tb(self.exc.__traceback__, file=output, show_lines=True)
+        result = output.getvalue()
+        self.assertIn('x = 1 / 0', result)
+        self.assertIn('File ', result)
+
+    def test_print_tb_show_lines_false(self):
+        """Test print_tb with show_lines=False"""
+        output = StringIO()
+        traceback.print_tb(self.exc.__traceback__, file=output, show_lines=False)
+        result = output.getvalue()
+        self.assertNotIn('x = 1 / 0', result)
+        self.assertIn('File ', result)  # File info should still be present
+
+    def test_format_tb_show_lines_true(self):
+        """Test format_tb with show_lines=True (default)"""
+        result = traceback.format_tb(self.exc.__traceback__, show_lines=True)
+        formatted = ''.join(result)
+        self.assertIn('x = 1 / 0', formatted)
+        self.assertIn('File ', formatted)
+
+    def test_format_tb_show_lines_false(self):
+        """Test format_tb with show_lines=False"""
+        result = traceback.format_tb(self.exc.__traceback__, show_lines=False)
+        formatted = ''.join(result)
+        self.assertNotIn('x = 1 / 0', formatted)
+        self.assertIn('File ', formatted)  # File info should still be present
+
+    def test_print_exception_show_lines_true(self):
+        """Test print_exception with show_lines=True (default)"""
+        output = StringIO()
+        traceback.print_exception(self.exc, file=output, show_lines=True)
+        result = output.getvalue()
+        self.assertIn('x = 1 / 0', result)
+        self.assertIn('ZeroDivisionError', result)
+
+    def test_print_exception_show_lines_false(self):
+        """Test print_exception with show_lines=False"""
+        output = StringIO()
+        traceback.print_exception(self.exc, file=output, show_lines=False)
+        result = output.getvalue()
+        self.assertNotIn('x = 1 / 0', result)
+        self.assertIn('ZeroDivisionError', result)  # Exception type should still be present
+
+    def test_format_exception_show_lines_true(self):
+        """Test format_exception with show_lines=True (default)"""
+        result = traceback.format_exception(self.exc, show_lines=True)
+        formatted = ''.join(result)
+        self.assertIn('x = 1 / 0', formatted)
+        self.assertIn('ZeroDivisionError', formatted)
+
+    def test_format_exception_show_lines_false(self):
+        """Test format_exception with show_lines=False"""
+        result = traceback.format_exception(self.exc, show_lines=False)
+        formatted = ''.join(result)
+        self.assertNotIn('x = 1 / 0', formatted)
+        self.assertIn('ZeroDivisionError', formatted)  # Exception type should still be present
+
+    def test_print_exc_show_lines_false(self):
+        """Test print_exc with show_lines=False"""
+        # Override sys.exception() to return our test exception
+        original_exception = sys.exception
+        sys.exception = lambda: self.exc
+        try:
+            output = StringIO()
+            traceback.print_exc(file=output, show_lines=False)
+            result = output.getvalue()
+            self.assertNotIn('x = 1 / 0', result)
+            self.assertIn('ZeroDivisionError', result)
+        finally:
+            sys.exception = original_exception
+
+    def test_format_exc_show_lines_false(self):
+        """Test format_exc with show_lines=False"""
+        # Override sys.exception() to return our test exception
+        original_exception = sys.exception
+        sys.exception = lambda: self.exc
+        try:
+            result = traceback.format_exc(show_lines=False)
+            self.assertNotIn('x = 1 / 0', result)
+            self.assertIn('ZeroDivisionError', result)
+        finally:
+            sys.exception = original_exception
+
+    def test_print_stack_show_lines_false(self):
+        """Test print_stack with show_lines=False"""
+        output = StringIO()
+        traceback.print_stack(file=output, show_lines=False)
+        result = output.getvalue()
+        # Should not contain source code lines
+        lines = result.split('\n')
+        # Filter out empty lines and check that remaining lines are just file/line info
+        non_empty_lines = [line for line in lines if line.strip()]
+        for line in non_empty_lines:
+            if line.strip():
+                self.assertTrue(line.strip().startswith('File ') or
+                              'in ' in line or
+                              line.strip() == 'traceback.print_stack(file=output, show_lines=False)')
+
+    def test_format_stack_show_lines_false(self):
+        """Test format_stack with show_lines=False"""
+        result = traceback.format_stack(show_lines=False)
+        formatted = ''.join(result)
+        # Should contain file information but not source code
+        self.assertIn('File ', formatted)
+        # Check that the source code of this test is not included
+        self.assertNotIn('traceback.format_stack(show_lines=False)', formatted)
+
+    def test_format_list_show_lines_false(self):
+        """Test format_list with show_lines=False"""
+        tb_list = traceback.extract_tb(self.exc.__traceback__)
+        result = traceback.format_list(tb_list, show_lines=False)
+        formatted = ''.join(result)
+        self.assertNotIn('x = 1 / 0', formatted)
+        self.assertIn('File ', formatted)  # File info should still be present
+
+    def test_print_list_show_lines_false(self):
+        """Test print_list with show_lines=False"""
+        tb_list = traceback.extract_tb(self.exc.__traceback__)
+        output = StringIO()
+        traceback.print_list(tb_list, file=output, show_lines=False)
+        result = output.getvalue()
+        self.assertNotIn('x = 1 / 0', result)
+        self.assertIn('File ', result)  # File info should still be present
+
+    def test_traceback_exception_show_lines_false(self):
+        """Test TracebackException with show_lines=False"""
+        te = traceback.TracebackException.from_exception(self.exc)
+        result = list(te.format(show_lines=False))
+        formatted = ''.join(result)
+        self.assertNotIn('x = 1 / 0', formatted)
+        self.assertIn('ZeroDivisionError', formatted)
+
+    def test_traceback_exception_print_show_lines_false(self):
+        """Test TracebackException.print with show_lines=False"""
+        te = traceback.TracebackException.from_exception(self.exc)
+        output = StringIO()
+        te.print(file=output, show_lines=False)
+        result = output.getvalue()
+        self.assertNotIn('x = 1 / 0', result)
+        self.assertIn('ZeroDivisionError', result)
+
+
 if __name__ == "__main__":
     unittest.main()