summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_fstring.py
diff options
context:
space:
mode:
authorsunmy2019 <59365878+sunmy2019@users.noreply.github.com>2023-10-05 14:08:42 (GMT)
committerGitHub <noreply@github.com>2023-10-05 14:08:42 (GMT)
commit2cb62c6437fa07e08b4778f7ab9baa5f16ac01f2 (patch)
treee2745301a1a835dbf56197fd929db545f43dc208 /Lib/test/test_fstring.py
parentcc389ef627b2a486ab89d9a11245bef48224efb1 (diff)
downloadcpython-2cb62c6437fa07e08b4778f7ab9baa5f16ac01f2.zip
cpython-2cb62c6437fa07e08b4778f7ab9baa5f16ac01f2.tar.gz
cpython-2cb62c6437fa07e08b4778f7ab9baa5f16ac01f2.tar.bz2
gh-110309: Prune empty constant in format specs (#110320)
Co-authored-by: blurb-it[bot] <43283697+blurb-it[bot]@users.noreply.github.com>
Diffstat (limited to 'Lib/test/test_fstring.py')
-rw-r--r--Lib/test/test_fstring.py48
1 files changed, 48 insertions, 0 deletions
diff --git a/Lib/test/test_fstring.py b/Lib/test/test_fstring.py
index 4f05a14..dd8c2dd 100644
--- a/Lib/test/test_fstring.py
+++ b/Lib/test/test_fstring.py
@@ -514,6 +514,54 @@ x = (
self.assertEqual(type(format_spec), ast.JoinedStr)
self.assertEqual(len(format_spec.values), 0)
+ def test_ast_fstring_format_spec(self):
+ expr = "f'{1:{name}}'"
+
+ mod = ast.parse(expr)
+ self.assertEqual(type(mod), ast.Module)
+ self.assertEqual(len(mod.body), 1)
+
+ fstring = mod.body[0].value
+ self.assertEqual(type(fstring), ast.JoinedStr)
+ self.assertEqual(len(fstring.values), 1)
+
+ fv = fstring.values[0]
+ self.assertEqual(type(fv), ast.FormattedValue)
+
+ format_spec = fv.format_spec
+ self.assertEqual(type(format_spec), ast.JoinedStr)
+ self.assertEqual(len(format_spec.values), 1)
+
+ format_spec_value = format_spec.values[0]
+ self.assertEqual(type(format_spec_value), ast.FormattedValue)
+ self.assertEqual(format_spec_value.value.id, 'name')
+
+ expr = "f'{1:{name1}{name2}}'"
+
+ mod = ast.parse(expr)
+ self.assertEqual(type(mod), ast.Module)
+ self.assertEqual(len(mod.body), 1)
+
+ fstring = mod.body[0].value
+ self.assertEqual(type(fstring), ast.JoinedStr)
+ self.assertEqual(len(fstring.values), 1)
+
+ fv = fstring.values[0]
+ self.assertEqual(type(fv), ast.FormattedValue)
+
+ format_spec = fv.format_spec
+ self.assertEqual(type(format_spec), ast.JoinedStr)
+ self.assertEqual(len(format_spec.values), 2)
+
+ format_spec_value = format_spec.values[0]
+ self.assertEqual(type(format_spec_value), ast.FormattedValue)
+ self.assertEqual(format_spec_value.value.id, 'name1')
+
+ format_spec_value = format_spec.values[1]
+ self.assertEqual(type(format_spec_value), ast.FormattedValue)
+ self.assertEqual(format_spec_value.value.id, 'name2')
+
+
def test_docstring(self):
def f():
f'''Not a docstring'''