X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/3501cefb09eb8448bd82287840c9093f10c25299..6ef3e466db7ad91dd6300d2412222ec912ae56e2:/src/black/__init__.py

diff --git a/src/black/__init__.py b/src/black/__init__.py
index f2efdec..d8b9819 100644
--- a/src/black/__init__.py
+++ b/src/black/__init__.py
@@ -1210,6 +1210,21 @@ def get_features_used(  # noqa: C901
                         if argch.type in STARS:
                             features.add(feature)
 
+        elif (
+            n.type in {syms.return_stmt, syms.yield_expr}
+            and len(n.children) >= 2
+            and n.children[1].type == syms.testlist_star_expr
+            and any(child.type == syms.star_expr for child in n.children[1].children)
+        ):
+            features.add(Feature.UNPACKING_ON_FLOW)
+
+        elif (
+            n.type == syms.annassign
+            and len(n.children) >= 4
+            and n.children[3].type == syms.testlist_star_expr
+        ):
+            features.add(Feature.ANN_ASSIGN_EXTENDED_RHS)
+
         # Python 2 only features (for its deprecation) except for integers, see above
         elif n.type == syms.print_stmt:
             features.add(Feature.PRINT_STMT)