Update test_fix_pyre_directives.py (#1082)

* Update test_fix_pyre_directives.py refactor with fstring to format string to make code more Pythonic. * Update test_fix_pyre_directives.py refactor with fstring to format string to make code more Pythonic. * Update test_fix_pyre_directives.py refactor with fstring to format string to make code more Pythonic. * Update test_fix_pyre_directives.py refactor with fstring to format string to make code more Pythonic. * Update test_fix_pyre_directives.py refactor with chain constant value assignment to make code more Pythonic * Update test_fix_pyre_directives.py refactor with chain constant value assignment to make code more Pythonic
Instagram · Jan 3, 2024 · dbbfe1e · dbbfe1e
1 parent 30df6fc
commit dbbfe1e
Showing 1 changed file with 9 additions and 29 deletions.
diff --git a/libcst/codemod/commands/tests/test_fix_pyre_directives.py b/libcst/codemod/commands/tests/test_fix_pyre_directives.py
@@ -14,14 +14,9 @@ def test_no_need_to_fix_simple(self) -> None:
         """
         Tests that a pyre-strict inside the module header doesn't get touched.
         """
-        before = """
-            # pyre-strict
-            from typing import List
-
-            def baz() -> List[Foo]:
-                pass
-        """
-        after = """
+        after = (
+            before
+        ) = """
             # pyre-strict
             from typing import List
 
@@ -34,16 +29,9 @@ def test_no_need_to_fix_complex_bottom(self) -> None:
         """
         Tests that a pyre-strict inside the module header doesn't get touched.
         """
-        before = """
-            # This is some header comment.
-            #
-            # pyre-strict
-            from typing import List
-
-            def baz() -> List[Foo]:
-                pass
-        """
-        after = """
+        after = (
+            before
+        ) = """
             # This is some header comment.
             #
             # pyre-strict
@@ -58,17 +46,9 @@ def test_no_need_to_fix_complex_top(self) -> None:
         """
         Tests that a pyre-strict inside the module header doesn't get touched.
         """
-        before = """
-            # pyre-strict
-            #
-            # This is some header comment.
-
-            from typing import List
-
-            def baz() -> List[Foo]:
-                pass
-        """
-        after = """
+        after = (
+            before
+        ) = """
             # pyre-strict
             #
             # This is some header comment.