test: added tests for functions with whitespaces in def nedbat#684

kajakaj · Dec 2, 2023 · cec888f · cec888f
1 parent fdb6841
commit cec888f
Showing 1 changed file with 50 additions and 0 deletions.
diff --git a/tests/test_coverage.py b/tests/test_coverage.py
@@ -1590,6 +1590,56 @@ def my_func_2(super_long_input_argument_0=0, super_long_input_argument_1=1, supe
             """,
             [1,5], "5", excludes=['my_func_2'])
 
+        self.check_coverage("""\
+            def my_func       (
+                super_long_input_argument_0=0,
+                super_long_input_argument_1=1,
+                super_long_input_argument_2=2):
+                pass
+
+            def my_func_2    (super_long_input_argument_0=0, super_long_input_argument_1=1, super_long_input_argument_2=2):
+                pass
+            """,
+            [1,5], "5", excludes=['my_func_2'])
+
+        self.check_coverage("""\
+            def my_func       (
+                super_long_input_argument_0=0,
+                super_long_input_argument_1=1,
+                super_long_input_argument_2=2):
+                pass
+
+            def my_func_2    (super_long_input_argument_0=0, super_long_input_argument_1=1, super_long_input_argument_2=2):
+                pass
+            """,
+            [], "5", excludes=['my_func'])
+
+        self.check_coverage("""\
+            def my_func \
+                (
+                super_long_input_argument_0=0,
+                super_long_input_argument_1=1
+                ):
+                pass
+
+            def my_func_2(super_long_input_argument_0=0, super_long_input_argument_1=1, super_long_input_argument_2=2):
+                pass
+            """,
+            [1,5], "5", excludes=['my_func_2'])
+
+        self.check_coverage("""\
+            def my_func \
+                (
+                super_long_input_argument_0=0,
+                super_long_input_argument_1=1
+                ):
+                pass
+
+            def my_func_2(super_long_input_argument_0=0, super_long_input_argument_1=1, super_long_input_argument_2=2):
+                pass
+            """,
+            [], "5", excludes=['my_func'])
+
     def test_excluding_method(self) -> None:
         self.check_coverage("""\
             class Fooey: