Soft comparison of --required-version (#2832)

[etc/vim.git] / tests / test_black.py

diff --git a/tests/test_black.py b/tests/test_black.py
index 2dd284f2cd6ed176f030e986f1aacaf13d4dc823..b04c0a66fe9e4120c69121c8fd7dbbf033dec4dd 100644 (file)
--- a/tests/test_black.py
+++ b/tests/test_black.py
@@ -1198,6 +1198,20 @@ class BlackTestCase(BlackBaseTestCase):
             ignore_config=True,
         )
 
+    def test_required_version_matches_partial_version(self) -> None:
+        self.invokeBlack(
+            ["--required-version", black.__version__.split(".")[0], "-c", "0"],
+            exit_code=0,
+            ignore_config=True,
+        )
+
+    def test_required_version_does_not_match_on_minor_version(self) -> None:
+        self.invokeBlack(
+            ["--required-version", black.__version__.split(".")[0] + ".999", "-c", "0"],
+            exit_code=1,
+            ignore_config=True,
+        )
+
     def test_required_version_does_not_match_version(self) -> None:
         result = BlackRunner().invoke(
             black.main,