Force parentheses for named expressions in more contexts (#6494)

See:
https://github.com/astral-sh/ruff/pull/6436#issuecomment-1673583888.

crates/ruff_python_formatter/resources/test/fixtures/ruff/expression/named_expr.py

@@ -36,3 +36,18 @@ except (e := Exception):
 (x := 1)
 
 (x := 1) + (y := 2)
+
+with (x := 1):
+    pass
+
+
+def f():
+    yield (x := 1)
+
+
+def f():
+    yield from (x := 1)
+
+
+async def f():
+    await (x := 1)

crates/ruff_python_formatter/src/expression/expr_named_expr.rs

@@ -44,6 +44,9 @@ impl NeedsParentheses for ExprNamedExpr {
             || parent.is_stmt_return()
             || parent.is_except_handler_except_handler()
             || parent.is_with_item()
+            || parent.is_expr_yield()
+            || parent.is_expr_yield_from()
+            || parent.is_expr_await()
             || parent.is_stmt_delete()
             || parent.is_stmt_for()
         {

crates/ruff_python_formatter/tests/snapshots/format@expression__named_expr.py.snap

@@ -42,6 +42,21 @@ except (e := Exception):
 (x := 1)
 
 (x := 1) + (y := 2)
+
+with (x := 1):
+    pass
+
+
+def f():
+    yield (x := 1)
+
+
+def f():
+    yield from (x := 1)
+
+
+async def f():
+    await (x := 1)
 ```
 
 ## Output
@@ -82,6 +97,21 @@ except (e := Exception):
 (x := 1)
 
 (x := 1) + (y := 2)
+
+with (x := 1):
+    pass
+
+
+def f():
+    yield (x := 1)
+
+
+def f():
+    yield from (x := 1)
+
+
+async def f():
+    await (x := 1)
 ```