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)