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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
# SPDX-FileCopyrightText: David Fritzsche
# SPDX-License-Identifier: CC0-1.0
from typing import Optional
import pytest
from pytest_mypy_testing.message import Message, Severity
@pytest.mark.parametrize(
"string,expected", [("r", Severity.NOTE), ("N", Severity.NOTE)]
)
def test_init_severity(string: str, expected: Severity):
assert Severity.from_string(string) == expected
@pytest.mark.parametrize(
"filename,comment,severity,message,error_code",
[
("z.py", "# E: bar", Severity.ERROR, "bar", None),
("z.py", "# E: bar", Severity.ERROR, "bar", "foo"),
("z.py", "# E: bar [foo]", Severity.ERROR, "bar", "foo"),
("z.py", "# E: bar [foo]", Severity.ERROR, "bar", ""),
("z.py", "#type:ignore# W: bar", Severity.WARNING, "bar", None),
("z.py", "# type: ignore # W: bar", Severity.WARNING, "bar", None),
("z.py", "# R: bar", Severity.NOTE, "Revealed type is 'bar'", None),
],
)
def test_message_from_comment(
filename: str,
comment: str,
severity: Severity,
message: str,
error_code: Optional[str],
):
lineno = 123
actual = Message.from_comment(filename, lineno, comment)
expected = Message(
filename=filename,
lineno=lineno,
colno=None,
severity=severity,
message=message,
error_code=error_code,
)
assert actual == expected
def test_message_from_invalid_comment():
with pytest.raises(ValueError):
Message.from_comment("foo.py", 1, "# fubar")
@pytest.mark.parametrize(
"line,severity,message",
[
("z.py:1: note: bar", Severity.NOTE, "bar"),
("z.py:1:2: note: bar", Severity.NOTE, "bar"),
("z.py:1:2: error: fubar", Severity.ERROR, "fubar"),
],
)
def test_message_from_output(line: str, severity: Severity, message: str):
msg = Message.from_output(line)
assert msg.message == message
assert msg.severity == severity
@pytest.mark.parametrize(
"output",
[
"foo.py:a: fubar",
"fubar",
"foo.py:1: fubar",
"foo.py:1:1: fubar",
"foo.py:1:1: not: fubar",
],
)
def test_message_from_invalid_output(output):
with pytest.raises(ValueError):
Message.from_output(output)
MSG_WITHOUT_COL = Message("z.py", 13, None, Severity.NOTE, "foo")
MSG_WITH_COL = Message("z.py", 13, 23, Severity.NOTE, "foo")
@pytest.mark.parametrize(
"a,b",
[
(MSG_WITH_COL, MSG_WITH_COL),
(MSG_WITH_COL, MSG_WITHOUT_COL),
(MSG_WITHOUT_COL, MSG_WITHOUT_COL),
(MSG_WITHOUT_COL, MSG_WITH_COL),
],
)
def test_message_eq(a: Message, b: Message):
assert a == b
assert not (a != b)
def test_message_neq_with_not_message():
assert MSG_WITH_COL != 23
assert MSG_WITH_COL != "abc"
def test_message_hash():
assert hash(MSG_WITH_COL) == hash(MSG_WITHOUT_COL)
def test_message_str():
assert str(MSG_WITHOUT_COL) == "z.py:13: note: foo"
assert str(MSG_WITH_COL) == "z.py:13:23: note: foo"
|
