add handling for future E204 whitespace after decorator

hhatto · Jun 15, 2024 · 13fe322 · 13fe322
1 parent a109296
commit 13fe322
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/autopep8.py b/autopep8.py
@@ -509,6 +509,7 @@ def __init__(self, filename,
         self.fix_e133 = self.fix_e131
         self.fix_e202 = self.fix_e201
         self.fix_e203 = self.fix_e201
+        self.fix_e204 = self.fix_e201
         self.fix_e211 = self.fix_e201
         self.fix_e221 = self.fix_e271
         self.fix_e222 = self.fix_e271

diff --git a/test/test_autopep8.py b/test/test_autopep8.py
@@ -1859,6 +1859,12 @@ def test_e203_with_newline(self):
         with autopep8_context(line, options=['--select=E203']) as result:
             self.assertEqual(fixed, result)
 
+    def test_e204(self):
+        line = '@ decorator\n'
+        fixed = '@decorator\n'
+        with autopep8_context(line, options=['--select=E204']) as result:
+            self.assertEqual(fixed, result)
+
     def test_e211(self):
         line = 'd = [1, 2, 3]\nprint(d  [0])\n'
         fixed = 'd = [1, 2, 3]\nprint(d[0])\n'