Update test_wemake.py

PyCQA · Feb 28, 2024 · b553962 · b553962
1 parent 1593706
commit b553962
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/tests/unit/profiles/test_wemake.py b/tests/unit/profiles/test_wemake.py
@@ -88,10 +88,22 @@ class _ClassVisitor(ast.NodeVisitor): ...
 
 
 def test_wemake_snippet_four():
-    """80 line length should be fixed"""
+    """80 line length should not be fixed"""
     wemake_isort_test(
         """
 from typing import Iterable, Iterator, Optional, Sequence, Tuple, TypeVar, Union
+""",
+        """
+from typing import Iterable, Iterator, Optional, Sequence, Tuple, TypeVar, Union
+""",
+    )
+
+
+def test_wemake_snippet_five():
+    """81 line length should be fixed"""
+    wemake_isort_test(
+        """
+from typing import Iterable, Iterator, Optional, Sequence, Tuple, TypeVar, Union1
 """,
         """
 from typing import (
@@ -101,7 +113,7 @@ def test_wemake_snippet_four():
     Sequence,
     Tuple,
     TypeVar,
-    Union,
+    Union1,
 )
 """,
     )