From: Pablo Galindo <Pablogsal@gmail.com>
Date: Wed, 18 Sep 2019 11:54:40 +0000 (+0100)
Subject: Support PEP 572 in while statements (#1028)
X-Git-Url: https://git.madduck.net/etc/vim.git/commitdiff_plain/673327449f86fce558adde153bb6cbe54bfebad2

Support PEP 572 in while statements (#1028)

Commit d8fa8df0526de9c0968e0a3568008f58eae45364 added support for
parsing and formatting PEP572, but it missed the posibility to add
PEP572 syntax in while statements.
---

diff --git a/blib2to3/Grammar.txt b/blib2to3/Grammar.txt
index daecb15..d90710f 100644
--- a/blib2to3/Grammar.txt
+++ b/blib2to3/Grammar.txt
@@ -108,7 +108,7 @@ assert_stmt: 'assert' test [',' test]
 compound_stmt: if_stmt | while_stmt | for_stmt | try_stmt | with_stmt | funcdef | classdef | decorated | async_stmt
 async_stmt: ASYNC (funcdef | with_stmt | for_stmt)
 if_stmt: 'if' namedexpr_test ':' suite ('elif' namedexpr_test ':' suite)* ['else' ':' suite]
-while_stmt: 'while' test ':' suite ['else' ':' suite]
+while_stmt: 'while' namedexpr_test ':' suite ['else' ':' suite]
 for_stmt: 'for' exprlist 'in' testlist ':' suite ['else' ':' suite]
 try_stmt: ('try' ':' suite
            ((except_clause ':' suite)+
diff --git a/tests/data/pep_572.py b/tests/data/pep_572.py
index 2b240be..563e8a7 100644
--- a/tests/data/pep_572.py
+++ b/tests/data/pep_572.py
@@ -38,3 +38,6 @@ if self._is_special and (ans := self._check_nans(context=context)):
 foo(b := 2, a=1)
 foo((b := 2), a=1)
 foo(c=(b := 2), a=1)
+
+while x:= f(x):
+    pass