Unpacking on flow constructs (return/yield) now implies 3.8+ (#2700)

[etc/vim.git] / src / black / __init__.py

diff --git a/src/black/__init__.py b/src/black/__init__.py
index f2efdec83b22a53d1ad00f6d1b1057dcae1f0c0f..08c239dc15561be1ceb41a7411957289880620ac 100644 (file)
--- a/src/black/__init__.py
+++ b/src/black/__init__.py
@@ -1210,6 +1210,14 @@ 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)
+
         # Python 2 only features (for its deprecation) except for integers, see above
         elif n.type == syms.print_stmt:
             features.add(Feature.PRINT_STMT)