add: unit test for e501 with async def

hhatto · Mar 14, 2024 · 47e37a1 · 47e37a1
1 parent 30e79c9
commit 47e37a1
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/test/test_autopep8.py b/test/test_autopep8.py
@@ -3709,6 +3709,19 @@ def foo(sldfkjlsdfsdf, kksdfsdfsf, sdfsdfsdf, sdfsdfkdk, szdfsdfsdf,
         with autopep8_context(line, options=['--aggressive']) as result:
             self.assertEqual(fixed, result)
 
+    def test_e501_aggressive_with_async_def(self):
+        line = """\
+async def foo(sldfkjlsdfsdf, kksdfsdfsf,sdfsdfsdf, sdfsdfkdk, szdfsdfsdf, sdfsdfsdfsdlkfjsdlf, sdfsdfddf,sdfsdfsfd, sdfsdfdsf):
+    pass
+"""
+        fixed = """\
+async def foo(sldfkjlsdfsdf, kksdfsdfsf, sdfsdfsdf, sdfsdfkdk,
+              szdfsdfsdf, sdfsdfsdfsdlkfjsdlf, sdfsdfddf, sdfsdfsfd, sdfsdfdsf):
+    pass
+"""
+        with autopep8_context(line, options=['--aggressive']) as result:
+            self.assertEqual(fixed, result)
+
     def test_e501_more_aggressive_with_def(self):
         line = """\
 def foobar(sldfkjlsdfsdf, kksdfsdfsf,sdfsdfsdf, sdfsdfkdk, szdfsdfsdf, sdfsdfsdfsdlkfjsdlf, sdfsdfddf,sdfsdfsfd, sdfsdfdsf):
@@ -3730,6 +3743,27 @@ def foobar(
         with autopep8_context(line, options=['-aa']) as result:
             self.assertEqual(fixed, result)
 
+    def test_e501_more_aggressive_with_async_def(self):
+        line = """\
+async def foobar(sldfkjlsdfsdf, kksdfsdfsf,sdfsdfsdf, sdfsdfkdk, szdfsdfsdf, sdfsdfsdfsdlkfjsdlf, sdfsdfddf,sdfsdfsfd, sdfsdfdsf):
+    pass
+"""
+        fixed = """\
+async def foobar(
+        sldfkjlsdfsdf,
+        kksdfsdfsf,
+        sdfsdfsdf,
+        sdfsdfkdk,
+        szdfsdfsdf,
+        sdfsdfsdfsdlkfjsdlf,
+        sdfsdfddf,
+        sdfsdfsfd,
+        sdfsdfdsf):
+    pass
+"""
+        with autopep8_context(line, options=['-aa']) as result:
+            self.assertEqual(fixed, result)
+
     def test_e501_aggressive_with_tuple(self):
         line = """\
 def f():