added strict mypy checks

This commit is contained in:
Brian Bjarke Jensen
2025-08-28 20:56:28 +02:00
parent cb036da997
commit 5e610dd469
6 changed files with 46 additions and 23 deletions
+18 -15
View File
@@ -3,12 +3,15 @@
import os
import unittest
from pytohn_utils import check_env
from python_utils import check_env
class TestCheckEnv(unittest.TestCase):
set_vars: list[str]
unset_var: str
@classmethod
def setUpClass(cls):
def setUpClass(cls) -> None:
"""Set up test environment variables."""
cls.set_vars = ["A", "B", "C"]
for var in cls.set_vars:
@@ -20,41 +23,41 @@ class TestCheckEnv(unittest.TestCase):
)
@classmethod
def tearDownClass(cls):
def tearDownClass(cls) -> None:
for var in cls.set_vars:
if var in os.environ:
del os.environ[var]
os.environ.pop(var, None)
def test_check_env_accepts_single_string(self):
def test_check_env_accepts_single_string(self) -> None:
"""Test check_env function with a valid single string."""
self.assertIsNone(check_env(self.set_vars[0]))
check_env(self.set_vars[0])
def test_check_env_accepts_list_of_strings(self):
def test_check_env_accepts_list_of_strings(self) -> None:
"""Test check_env function with a valid list of strings."""
self.assertIsNone(check_env(self.set_vars))
check_env(self.set_vars)
def test_check_env_accepts_set_of_strings(self):
def test_check_env_accepts_set_of_strings(self) -> None:
"""Test check_env function with a valid set of strings."""
self.assertIsNone(check_env(set(self.set_vars)))
check_env(set(self.set_vars))
def test_check_env_accepts_tuple_of_strings(self):
def test_check_env_accepts_tuple_of_strings(self) -> None:
"""Test check_env function with a valid tuple of strings."""
self.assertIsNone(check_env(tuple(self.set_vars)))
check_env(tuple(self.set_vars))
def test_check_env_rejects_empty_list(self):
def test_check_env_rejects_empty_list(self) -> None:
"""Test check_env function with an empty list."""
with self.assertRaises(ValueError):
check_env([])
def test_check_env_raises_error_on_missing_vars(self):
def test_check_env_raises_error_on_missing_vars(self) -> None:
"""Test check_env function with missing environment variables."""
with self.assertRaises(OSError):
check_env(self.unset_var)
def test_check_env_does_not_raise_error_when_missing_ok_flag_set(self):
def test_check_env_does_not_raise_error_when_missing_ok_flag_set(self) -> None:
"""Test that check_env does not raise an error when the environment variable is missing and missing_ok=True.
The 'missing_ok' flag allows the function to skip raising an error if the specified environment variable(s) are not set.
"""
self.assertIsNone(check_env(self.unset_var, missing_ok=True))
check_env(self.unset_var, missing_ok=True)
+6 -1
View File
@@ -1,6 +1,7 @@
"""Unittests for the readable function."""
import pytest
from typing import Any
from python_utils import readable_unit
@@ -31,7 +32,11 @@ TEST_CASES = [
@pytest.mark.parametrize("value, unit, expected", TEST_CASES)
def test_readable_unit(value, unit, expected):
def test_readable_unit(
value: Any, # Allow invalid types for testing
unit: Any, # Allow invalid types for testing
expected: Any, # Allow invalid types for testing
) -> None:
"""Test readable_unit function with valid inputs."""
if isinstance(expected, type) and issubclass(expected, Exception):
with pytest.raises(expected):