add tests about parenthesized context managers

psf · Nov 13, 2021 · d16f6c7 · d16f6c7
1 parent 126a522
commit d16f6c7
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/tests/data/parenthesized_context_managers.py b/tests/data/parenthesized_context_managers.py
@@ -0,0 +1,21 @@
+with (CtxManager() as example):
+    ...
+
+with (CtxManager1(), CtxManager2()):
+    ...
+
+with (CtxManager1() as example, CtxManager2()):
+    ...
+
+with (CtxManager1(), CtxManager2() as example):
+    ...
+
+with (CtxManager1() as example1, CtxManager2() as example2):
+    ...
+
+with (
+    CtxManager1() as example1,
+    CtxManager2() as example2,
+    CtxManager3() as example3,
+):
+    ...
diff --git a/tests/test_format.py b/tests/test_format.py
@@ -70,7 +70,11 @@
     "percent_precedence",
 ]
 
-PY310_CASES = ["pattern_matching_simple", "pattern_matching_complex"]
+PY310_CASES = [
+    "pattern_matching_simple",
+    "pattern_matching_complex",
+    "parenthesized_context_managers",
+]
 
 SOURCES = [
     "src/black/__init__.py",