first comit

venv/lib/python3.10/site-packages/zmq/tests/test_mypy.py

@@ -0,0 +1,73 @@
+"""
+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))