mirror of https://github.com/zeromq/pyzmq.git
74 lines
1.6 KiB
Python
74 lines
1.6 KiB
Python
"""
|
|
Test our typing with mypy
|
|
"""
|
|
import os
|
|
import sys
|
|
from subprocess import PIPE, STDOUT, Popen
|
|
|
|
import pytest
|
|
|
|
import zmq
|
|
|
|
pytest.importorskip("mypy")
|
|
|
|
zmq_dir = os.path.dirname(zmq.__file__)
|
|
|
|
|
|
def resolve_repo_dir(path):
|
|
"""Resolve a dir in the repo
|
|
|
|
Resolved relative to zmq dir
|
|
|
|
fallback on CWD (e.g. test run from repo, zmq installed, not -e)
|
|
"""
|
|
resolved_path = os.path.join(os.path.dirname(zmq_dir), path)
|
|
|
|
# fallback on CWD
|
|
if not os.path.exists(resolved_path):
|
|
resolved_path = path
|
|
return resolved_path
|
|
|
|
|
|
examples_dir = resolve_repo_dir("examples")
|
|
mypy_dir = resolve_repo_dir("mypy_tests")
|
|
|
|
|
|
def run_mypy(*mypy_args):
|
|
"""Run mypy for a path
|
|
|
|
Captures output and reports it on errors
|
|
"""
|
|
p = Popen(
|
|
[sys.executable, "-m", "mypy"] + list(mypy_args), stdout=PIPE, stderr=STDOUT
|
|
)
|
|
o, _ = p.communicate()
|
|
out = o.decode("utf8", "replace")
|
|
print(out)
|
|
assert p.returncode == 0, out
|
|
|
|
|
|
if os.path.exists(examples_dir):
|
|
examples = [
|
|
d
|
|
for d in os.listdir(examples_dir)
|
|
if os.path.isdir(os.path.join(examples_dir, d))
|
|
]
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
not os.path.exists(examples_dir), reason="only test from examples directory"
|
|
)
|
|
@pytest.mark.parametrize("example", examples)
|
|
def test_mypy_example(example):
|
|
example_dir = os.path.join(examples_dir, example)
|
|
run_mypy("--disallow-untyped-calls", example_dir)
|
|
|
|
|
|
if os.path.exists(mypy_dir):
|
|
mypy_tests = [p for p in os.listdir(mypy_dir) if p.endswith(".py")]
|
|
|
|
|
|
@pytest.mark.parametrize("filename", mypy_tests)
|
|
def test_mypy(filename):
|
|
run_mypy("--disallow-untyped-calls", os.path.join(mypy_dir, filename))
|