test only in 3.8+

psf · Dec 10, 2022 · 8310b81 · 8310b81
1 parent bae256b
commit 8310b81
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 9 deletions.
diff --git a/tests/data/preview/return_annotation_brackets.py b/tests/data/preview/return_annotation_brackets.py
@@ -91,10 +91,6 @@ def foo() -> tuple[int, int, int,]:
 def frobnicate() -> "ThisIsTrulyUnreasonablyExtremelyLongClassName | list[ThisIsTrulyUnreasonablyExtremelyLongClassName]":
     pass
 
-# Should not remove parens around walrus operator
-def walrus() -> (x := 1):
-    pass
-
 # output
 # Control
 def double(a: int) -> int:
@@ -224,8 +220,3 @@ def frobnicate() -> (
     " list[ThisIsTrulyUnreasonablyExtremelyLongClassName]"
 ):
     pass
-
-
-# Should not remove parens around walrus operator
-def walrus() -> (x := 1):
-    pass
diff --git a/tests/data/preview_38/return_annotation_brackets_py38.py b/tests/data/preview_38/return_annotation_brackets_py38.py
@@ -0,0 +1,9 @@
+# Should not remove parens around walrus operator
+def walrus() -> (x := 1):
+    pass
+
+
+# output
+# Should not remove parens around walrus operator
+def walrus() -> (x := 1):
+    pass
diff --git a/tests/test_format.py b/tests/test_format.py
@@ -44,6 +44,13 @@ def test_preview_format(filename: str) -> None:
     )
 
 
+@pytest.mark.parametrize("filename", all_data_cases("preview_38"))
+def test_preview_minimum_python_38_format(filename: str) -> None:
+    source, expected = read_data("preview_38", filename)
+    mode = black.Mode(preview=True)
+    assert_format(source, expected, mode, minimum_version=(3, 9))
+
+
 @pytest.mark.parametrize("filename", all_data_cases("preview_39"))
 def test_preview_minimum_python_39_format(filename: str) -> None:
     source, expected = read_data("preview_39", filename)