Fix feature detection for positional-only arguments in lambdas (#2532)

diff --git a/CHANGES.md b/CHANGES.md
index f5b11de48032bfb2712a52593b3495bfe1ef76a4..864f0a54410b33cadeaa8ee9b2a5d3817aa9c9d1 100644 (file)
--- a/CHANGES.md
+++ b/CHANGES.md
@@ -5,6 +5,7 @@
 ### _Black_
 
 - Add new `--workers` parameter (#2514)
 ### _Black_
 
 - Add new `--workers` parameter (#2514)

+- Fixed feature detection for positional-only arguments in lambdas (#2532)

 - Bumped typed-ast version minimum to 1.4.3 for 3.10 compatiblity (#2519)
 
 ### _Blackd_
 - Bumped typed-ast version minimum to 1.4.3 for 3.10 compatiblity (#2519)
 
 ### _Blackd_

diff --git a/src/black/__init__.py b/src/black/__init__.py
index 83a39234d38d1fd0ac3e506f187b022cb31d542b..fdbaf040d642b0a41d15d0f98112090971425643 100644 (file)
--- a/src/black/__init__.py
+++ b/src/black/__init__.py
@@ -1123,7 +1123,11 @@ def get_features_used(node: Node) -> Set[Feature]:
                 features.add(Feature.NUMERIC_UNDERSCORES)
 
         elif n.type == token.SLASH:
                 features.add(Feature.NUMERIC_UNDERSCORES)
 
         elif n.type == token.SLASH:

-            if n.parent and n.parent.type in {syms.typedargslist, syms.arglist}:
+            if n.parent and n.parent.type in {
+                syms.typedargslist,
+                syms.arglist,
+                syms.varargslist,
+            }:

                 features.add(Feature.POS_ONLY_ARGUMENTS)
 
         elif n.type == token.COLONEQUAL:
                 features.add(Feature.POS_ONLY_ARGUMENTS)
 
         elif n.type == token.COLONEQUAL:

diff --git a/tests/test_black.py b/tests/test_black.py
index f25db1b73d1ef9b102cce37ae1c89fe3566efbb4..beb56cf1fdb24bde7256366f0aa1ebf7acb23f60 100644 (file)
--- a/tests/test_black.py
+++ b/tests/test_black.py
@@ -805,6 +805,10 @@ class BlackTestCase(BlackBaseTestCase):
         self.assertEqual(black.get_features_used(node), set())
         node = black.lib2to3_parse(expected)
         self.assertEqual(black.get_features_used(node), set())
         self.assertEqual(black.get_features_used(node), set())
         node = black.lib2to3_parse(expected)
         self.assertEqual(black.get_features_used(node), set())

+        node = black.lib2to3_parse("lambda a, /, b: ...")
+        self.assertEqual(black.get_features_used(node), {Feature.POS_ONLY_ARGUMENTS})
+        node = black.lib2to3_parse("def fn(a, /, b): ...")
+        self.assertEqual(black.get_features_used(node), {Feature.POS_ONLY_ARGUMENTS})

 
     def test_get_future_imports(self) -> None:
         node = black.lib2to3_parse("\n")
 
     def test_get_future_imports(self) -> None:
         node = black.lib2to3_parse("\n")