https://github.com/python/cpython/commit/0a26aa5007cb32610366c31fbac846b5fe2f4f90
commit: 0a26aa5007cb32610366c31fbac846b5fe2f4f90
branch: main
author: Tomas R <[email protected]>
committer: encukou <[email protected]>
date: 2024-07-13T14:59:15+02:00
summary:
gh-121671: Increase test coverage of `ast.get_docstring` (GH-121674)
Increase test coverage for `ast.get_docstring`
files:
M Lib/test/test_ast.py
diff --git a/Lib/test/test_ast.py b/Lib/test/test_ast.py
index 497c3f261a1fca..55725ec36fd3a7 100644
--- a/Lib/test/test_ast.py
+++ b/Lib/test/test_ast.py
@@ -1821,6 +1821,12 @@ def test_get_docstring(self):
node = ast.parse('async def foo():\n """spam\n ham"""')
self.assertEqual(ast.get_docstring(node.body[0]), 'spam\nham')
+ node = ast.parse('async def foo():\n """spam\n ham"""')
+ self.assertEqual(ast.get_docstring(node.body[0], clean=False), 'spam\n
ham')
+
+ node = ast.parse('x')
+ self.assertRaises(TypeError, ast.get_docstring, node.body[0])
+
def test_get_docstring_none(self):
self.assertIsNone(ast.get_docstring(ast.parse('')))
node = ast.parse('x = "not docstring"')
@@ -1845,6 +1851,9 @@ def test_get_docstring_none(self):
node = ast.parse('async def foo():\n x = "not docstring"')
self.assertIsNone(ast.get_docstring(node.body[0]))
+ node = ast.parse('async def foo():\n 42')
+ self.assertIsNone(ast.get_docstring(node.body[0]))
+
def test_multi_line_docstring_col_offset_and_lineno_issue16806(self):
node = ast.parse(
'"""line one\nline two"""\n\n'
_______________________________________________
Python-checkins mailing list -- [email protected]
To unsubscribe send an email to [email protected]
https://mail.python.org/mailman3/lists/python-checkins.python.org/
Member address: [email protected]
