X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/f10ce0c942b41cd4c6802ba690a432c6adedc05e..dc90d4951f66ac665582159537b902017d9a0361:/tests/test_black.py

diff --git a/tests/test_black.py b/tests/test_black.py
index 63cd716..8726cc1 100644
--- a/tests/test_black.py
+++ b/tests/test_black.py
@@ -810,6 +810,14 @@ class BlackTestCase(BlackBaseTestCase):
         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})
+        node = black.lib2to3_parse("def fn(): yield a, b")
+        self.assertEqual(black.get_features_used(node), set())
+        node = black.lib2to3_parse("def fn(): return a, b")
+        self.assertEqual(black.get_features_used(node), set())
+        node = black.lib2to3_parse("def fn(): yield *b, c")
+        self.assertEqual(black.get_features_used(node), {Feature.UNPACKING_ON_FLOW})
+        node = black.lib2to3_parse("def fn(): return a, *b, c")
+        self.assertEqual(black.get_features_used(node), {Feature.UNPACKING_ON_FLOW})
 
     def test_get_features_used_for_future_flags(self) -> None:
         for src, features in [