#! /usr/bin/env python3 # $Id: test_code_long.py 10022 2025-03-07 23:05:06Z aa-turner $ # Author: Guenter Milde # Copyright: This module has been placed in the public domain. """ Test the 'code' directive in body.py with syntax_highlight = 'long'. """ from pathlib import Path import sys import unittest if __name__ == '__main__': # prepend the "docutils root" to the Python library path # so we import the local `docutils` package. sys.path.insert(0, str(Path(__file__).resolve().parents[4])) from docutils.frontend import get_default_settings from docutils.parsers.rst import Parser from docutils.utils import new_document from docutils.utils.code_analyzer import with_pygments from test.test_parsers.test_rst.test_directives.test_code \ import PYGMENTS_2_14_PLUS, def_ws @unittest.skipUnless(with_pygments, 'needs Pygments') class ParserTestCase(unittest.TestCase): def test_parser(self): parser = Parser() settings = get_default_settings(Parser) settings.warning_stream = '' settings.syntax_highlight = 'long' for name, cases in totest.items(): for casenum, (case_input, case_expected) in enumerate(cases): with self.subTest(id=f'totest[{name!r}][{casenum}]'): document = new_document('test data', settings.copy()) parser.parse(case_input, document) output = document.pformat() self.assertEqual(case_expected, output) totest = {} totest['code_parsing_long'] = [ ["""\ .. code:: python3 :number-lines: 7 def my_function(): '''Test the lexer. ''' # and now for something completely different print(8/2) """, f"""\ 7 \n\ def {def_ws} my_function (): \n\ 8 \n\ \n\ \'\'\'Test the lexer. 9 \n\ \'\'\' \n\ 10 \n\ \n\ 11 \n\ \n\ # and now for something completely different \n\ 12 \n\ \n\ print ( 8 / 2 ) """ if PYGMENTS_2_14_PLUS else """\ 7 \n\ def \n\ my_function (): \n\ 8 \n\ \n\ \'\'\'Test the lexer. 9 \n\ \'\'\' \n\ 10 \n\ \n\ 11 \n\ \n\ # and now for something completely different \n\ 12 \n\ \n\ print ( 8 / 2 ) """], ["""\ .. code:: latex hello \\emph{world} % emphasize """, """\ hello \n\ \\emph { world } \n\ % emphasize """], ] if __name__ == '__main__': unittest.main()