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
|
import os
import re
import pytest
pytestmark = pytest.mark.mypy
api = pytest.importorskip("mypy.api")
PATH = os.path.join(os.path.dirname(__file__), "basic_typing_examples.py")
def get_lineno_type(line):
assert "revealed type" in line.lower()
_, lineno, msg = line.split(":", 2)
lineno = int(lineno)
pat = re.search("[\"'](.*)[\"']", msg)
typ = pat.groups()[0]
return lineno, typ
def test_mypy():
with open(PATH, "r") as fil:
ex_lines = fil.readlines()
stdout, stderr, code = api.run([PATH])
lines = stdout.splitlines()
for line in lines:
if "revealed type" in line.lower():
lineno, typ = get_lineno_type(line)
check = ex_lines[lineno - 1].split("#")[1].strip()
try:
exec(check, {"typ": typ})
except Exception:
assert (
False
), f"Failed check at {PATH}:{lineno}: {check!r}, where 'typ' is {typ!r}"
elif "success" not in line.lower():
assert False, line
|
