PyObject *filename,
int optimize,
int ff_features,
- int syntax_check_only);
+ int syntax_check_only,
+ int enable_warnings);
typedef struct {
r"Exceeds the limit \(\d+ digits\)"):
repr(ast.Constant(value=eval(source)))
- def test_pep_765_warnings(self):
- srcs = [
- textwrap.dedent("""
- def f():
- try:
- pass
- finally:
- return 42
- """),
- textwrap.dedent("""
- for x in y:
- try:
- pass
- finally:
- break
- """),
- textwrap.dedent("""
- for x in y:
- try:
- pass
- finally:
- continue
- """),
- ]
- for src in srcs:
- with self.assertWarnsRegex(SyntaxWarning, 'finally'):
- ast.parse(src)
-
- def test_pep_765_no_warnings(self):
- srcs = [
- textwrap.dedent("""
- try:
- pass
- finally:
- def f():
- return 42
- """),
- textwrap.dedent("""
- try:
- pass
- finally:
- for x in y:
- break
- """),
- textwrap.dedent("""
- try:
- pass
- finally:
- for x in y:
- continue
- """),
- ]
- for src in srcs:
- ast.parse(src)
-
def test_tstring(self):
# Test AST structure for simple t-string
tree = ast.parse('t"Hello"')
self.assertEqual(wm.category, SyntaxWarning)
self.assertIn("\"is\" with 'int' literal", str(wm.message))
+ @support.subTests('src', [
+ textwrap.dedent("""
+ def f():
+ try:
+ pass
+ finally:
+ return 42
+ """),
+ textwrap.dedent("""
+ for x in y:
+ try:
+ pass
+ finally:
+ break
+ """),
+ textwrap.dedent("""
+ for x in y:
+ try:
+ pass
+ finally:
+ continue
+ """),
+ ])
+ def test_pep_765_warnings(self, src):
+ with self.assertWarnsRegex(SyntaxWarning, 'finally'):
+ compile(src, '<string>', 'exec')
+ with warnings.catch_warnings():
+ warnings.simplefilter("error")
+ tree = ast.parse(src)
+ with self.assertWarnsRegex(SyntaxWarning, 'finally'):
+ compile(tree, '<string>', 'exec')
+
+ @support.subTests('src', [
+ textwrap.dedent("""
+ try:
+ pass
+ finally:
+ def f():
+ return 42
+ """),
+ textwrap.dedent("""
+ try:
+ pass
+ finally:
+ for x in y:
+ break
+ """),
+ textwrap.dedent("""
+ try:
+ pass
+ finally:
+ for x in y:
+ continue
+ """),
+ ])
+ def test_pep_765_no_warnings(self, src):
+ with warnings.catch_warnings():
+ warnings.simplefilter("error")
+ compile(src, '<string>', 'exec')
+
class TestBooleanExpression(unittest.TestCase):
class Value:
import contextlib
import io
+import warnings
import unittest
from unittest.mock import patch
from textwrap import dedent
code = "if foo:"
console = InteractiveColoredConsole(namespace, filename="<stdin>")
self.assertTrue(_more_lines(console, code))
+
+
+class TestWarnings(unittest.TestCase):
+ def test_pep_765_warning(self):
+ """
+ Test that a SyntaxWarning emitted from the
+ AST optimizer is only shown once in the REPL.
+ """
+ # gh-131927
+ console = InteractiveColoredConsole()
+ code = dedent("""\
+ def f():
+ try:
+ return 1
+ finally:
+ return 2
+ """)
+
+ with warnings.catch_warnings(record=True) as caught:
+ warnings.simplefilter("always")
+ console.runsource(code)
+
+ count = sum("'return' in a 'finally' block" in str(w.message)
+ for w in caught)
+ self.assertEqual(count, 1)
--- /dev/null
+:func:`ast.parse` no longer emits syntax warnings for
+``return``/``break``/``continue`` in ``finally`` (see :pep:`765`) -- they are
+only emitted during compilation.
int optimize;
int ff_features;
int syntax_check_only;
+ int enable_warnings;
_Py_c_array_t cf_finally; /* context for PEP 765 check */
int cf_finally_used;
static int
before_return(_PyASTPreprocessState *state, stmt_ty node_)
{
- if (state->cf_finally_used > 0) {
+ if (state->enable_warnings && state->cf_finally_used > 0) {
ControlFlowInFinallyContext *ctx = get_cf_finally_top(state);
if (ctx->in_finally && ! ctx->in_funcdef) {
if (!control_flow_in_finally_warning("return", node_, state)) {
static int
before_loop_exit(_PyASTPreprocessState *state, stmt_ty node_, const char *kw)
{
- if (state->cf_finally_used > 0) {
+ if (state->enable_warnings && state->cf_finally_used > 0) {
ControlFlowInFinallyContext *ctx = get_cf_finally_top(state);
if (ctx->in_finally && ! ctx->in_loop) {
if (!control_flow_in_finally_warning(kw, node_, state)) {
int
_PyAST_Preprocess(mod_ty mod, PyArena *arena, PyObject *filename, int optimize,
- int ff_features, int syntax_check_only)
+ int ff_features, int syntax_check_only, int enable_warnings)
{
_PyASTPreprocessState state;
memset(&state, 0, sizeof(_PyASTPreprocessState));
state.optimize = optimize;
state.ff_features = ff_features;
state.syntax_check_only = syntax_check_only;
+ state.enable_warnings = enable_warnings;
if (_Py_CArray_Init(&state.cf_finally, sizeof(ControlFlowInFinallyContext), 20) < 0) {
return -1;
}
c->c_optimize = (optimize == -1) ? _Py_GetConfig()->optimization_level : optimize;
c->c_save_nested_seqs = false;
- if (!_PyAST_Preprocess(mod, arena, filename, c->c_optimize, merged, 0)) {
+ if (!_PyAST_Preprocess(mod, arena, filename, c->c_optimize, merged, 0, 1)) {
return ERROR;
}
c->c_st = _PySymtable_Build(mod, filename, &c->c_future);
if (optimize == -1) {
optimize = _Py_GetConfig()->optimization_level;
}
- if (!_PyAST_Preprocess(mod, arena, filename, optimize, flags, no_const_folding)) {
+ if (!_PyAST_Preprocess(mod, arena, filename, optimize, flags, no_const_folding, 0)) {
return -1;
}
return 0;