fix: mypy

Author: BoboTiG

src/tests/test_setup.py

@@ -89,9 +89,11 @@ def test_sdist() -> None:
         f"mss-{__version__}/src/tests/test_macos.py",
         f"mss-{__version__}/src/tests/test_save.py",
         f"mss-{__version__}/src/tests/test_setup.py",
-        f"mss-{__version__}/src/tests/test_third_party.py",
         f"mss-{__version__}/src/tests/test_tools.py",
         f"mss-{__version__}/src/tests/test_windows.py",
+        f"mss-{__version__}/src/tests/third_party/__init__.py",
+        f"mss-{__version__}/src/tests/third_party/test_numpy.py",
+        f"mss-{__version__}/src/tests/third_party/test_pil.py",
     ]
 
 

src/tests/third_party/__init__.py

@@ -0,0 +1,21 @@
+"""This is part of the MSS Python's module.
+Source: https://github.com/BoboTiG/python-mss.
+"""
+
+import os
+import os.path
+
+import pytest
+
+from mss import mss
+
+pytest.importorskip("numpy", reason="Numpy module not available.")
+
+import numpy as np  # noqa: E402
+
+
+def test_numpy(pixel_ratio: int) -> None:
+    box = {"top": 0, "left": 0, "width": 10, "height": 10}
+    with mss(display=os.getenv("DISPLAY")) as sct:
+        img = np.array(sct.grab(box))
+    assert len(img) == 10 * pixel_ratio

src/tests/third_party/test_numpy.py

@@ -10,27 +10,11 @@
 
 from mss import mss
 
-try:
-    import numpy as np
-except (ImportError, RuntimeError):
-    # RuntimeError on Python 3.9 (macOS): Polyfit sanity test emitted a warning, ...
-    np = None
+pytest.importorskip("PIL", reason="PIL module not available.")
 
-try:
-    from PIL import Image
-except ImportError:
-    Image = None
+from PIL import Image  # noqa: E402
 
 
-@pytest.mark.skipif(np is None, reason="Numpy module not available.")
-def test_numpy(pixel_ratio: int) -> None:
-    box = {"top": 0, "left": 0, "width": 10, "height": 10}
-    with mss(display=os.getenv("DISPLAY")) as sct:
-        img = np.array(sct.grab(box))
-    assert len(img) == 10 * pixel_ratio
-
-
-@pytest.mark.skipif(Image is None, reason="PIL module not available.")
 def test_pil() -> None:
     width, height = 16, 16
     box = {"top": 0, "left": 0, "width": width, "height": height}
@@ -48,7 +32,6 @@ def test_pil() -> None:
     assert os.path.isfile("box.png")
 
 
-@pytest.mark.skipif(Image is None, reason="PIL module not available.")
 def test_pil_bgra() -> None:
     width, height = 16, 16
     box = {"top": 0, "left": 0, "width": width, "height": height}
@@ -66,7 +49,6 @@ def test_pil_bgra() -> None:
     assert os.path.isfile("box-bgra.png")
 
 
-@pytest.mark.skipif(Image is None, reason="PIL module not available.")
 def test_pil_not_16_rounded() -> None:
     width, height = 10, 10
     box = {"top": 0, "left": 0, "width": width, "height": height}