Add support for pyi files (#210)

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

diff --git a/tests/function2.py b/tests/function2.py
index 1b9d7b64a0476cc2212452fb615cf8ddffa503cc..e262e0588c8373e4cbab50f389e4c2c5d6c48db9 100644 (file)
--- a/tests/function2.py
+++ b/tests/function2.py
@@ -2,6 +2,11 @@ def f(
   a,
   **kwargs,
 ) -> A:
   a,
   **kwargs,
 ) -> A:

+    with cache_dir():
+        if something:
+            result = (
+                CliRunner().invoke(black.main, [str(src1), str(src2), "--diff", "--check"])
+            )

     return A(
         very_long_argument_name1=very_long_value_for_the_argument,
         very_long_argument_name2=very_long_value_for_the_argument,
     return A(
         very_long_argument_name1=very_long_value_for_the_argument,
         very_long_argument_name2=very_long_value_for_the_argument,


@@ -11,6 +16,11 @@ def f(
 # output
 
 def f(a, **kwargs) -> A:
 # output
 
 def f(a, **kwargs) -> A:

+    with cache_dir():
+        if something:
+            result = CliRunner().invoke(
+                black.main, [str(src1), str(src2), "--diff", "--check"]
+            )

     return A(
         very_long_argument_name1=very_long_value_for_the_argument,
         very_long_argument_name2=very_long_value_for_the_argument,
     return A(
         very_long_argument_name1=very_long_value_for_the_argument,
         very_long_argument_name2=very_long_value_for_the_argument,