1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
|
# Copyright 2017-2020 Palantir Technologies, Inc.
# Copyright 2021- Python Language Server Contributors.
import sys
from pylsp import lsp, uris
from pylsp.plugins import pyflakes_lint
from pylsp.workspace import Document
DOC_URI = uris.from_fs_path(__file__)
DOC = """import sys
def hello():
\tpass
import json
"""
DOC_SYNTAX_ERR = """def hello()
pass
"""
DOC_UNDEFINED_NAME_ERR = "a = b"
DOC_ENCODING = """# encoding=utf-8
import sys
"""
def test_pyflakes(workspace) -> None:
doc = Document(DOC_URI, workspace, DOC)
diags = pyflakes_lint.pylsp_lint(workspace, doc)
# One we're expecting is:
msg = "'sys' imported but unused"
unused_import = [d for d in diags if d["message"] == msg][0]
assert unused_import["range"]["start"] == {"line": 0, "character": 0}
assert unused_import["severity"] == lsp.DiagnosticSeverity.Warning
def test_syntax_error_pyflakes(workspace) -> None:
doc = Document(DOC_URI, workspace, DOC_SYNTAX_ERR)
diag = pyflakes_lint.pylsp_lint(workspace, doc)[0]
if sys.version_info[:2] >= (3, 10):
assert diag["message"] == "expected ':'"
else:
assert diag["message"] == "invalid syntax"
assert diag["range"]["start"] == {"line": 0, "character": 12}
assert diag["severity"] == lsp.DiagnosticSeverity.Error
def test_undefined_name_pyflakes(workspace) -> None:
doc = Document(DOC_URI, workspace, DOC_UNDEFINED_NAME_ERR)
diag = pyflakes_lint.pylsp_lint(workspace, doc)[0]
assert diag["message"] == "undefined name 'b'"
assert diag["range"]["start"] == {"line": 0, "character": 4}
assert diag["severity"] == lsp.DiagnosticSeverity.Error
def test_unicode_encoding(workspace) -> None:
doc = Document(DOC_URI, workspace, DOC_ENCODING)
diags = pyflakes_lint.pylsp_lint(workspace, doc)
assert len(diags) == 1
assert diags[0]["message"] == "'sys' imported but unused"
|