Added quick fix for when flake8 triggers the "multiple spaces after k…

…eyword" warning (#76)
microsoft · Nov 8, 2022 · 064d56e · 064d56e
1 parent 41fcabe
commit 064d56e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/bundled/tool/server.py b/bundled/tool/server.py
@@ -236,7 +236,7 @@ def code_action(params: lsp.CodeActionParams) -> List[lsp.CodeAction]:
     return code_actions
 
 
-@QUICK_FIXES.quick_fix(codes=["E273", "E274", "E275"])
+@QUICK_FIXES.quick_fix(codes=["E271", "E273", "E274", "E275"])
 def fix_format(
     _document: workspace.Document, diagnostics: List[lsp.Diagnostic]
 ) -> List[lsp.CodeAction]:

diff --git a/src/test/python_tests/test_code_actions.py b/src/test/python_tests/test_code_actions.py
@@ -19,6 +19,15 @@
 @pytest.mark.parametrize(
     ("code", "contents", "command"),
     [
+        (
+            "E271",
+            "from collections import    (namedtuple, defaultdict)",
+            {
+                "title": f"{LINTER}: Run document formatting",
+                "command": "editor.action.formatDocument",
+                "arguments": None,
+            },
+        ),
         (
             "E273",
             "x = 1 in\t[1, 2, 3]",