Fix missing coverage

src/djpress/__init__.py

@@ -5,12 +5,18 @@
 try:
     import tomllib
 except ImportError:
-    import tomli as tomllib
+    import tomli as tomllib  # Alias tomli as tomllib for compatibility with Python versions < 3.11
 
-# Define the path to pyproject.toml
-pyproject_path = Path(__file__).parent.parent.parent / "pyproject.toml"
 
-# Read the version from pyproject.toml
-with pyproject_path.open("rb") as f:
-    pyproject_data = tomllib.load(f)
-    __version__ = pyproject_data["project"]["version"]
+def load_version() -> str:
+    """Load the version from pyproject.toml."""
+    # Define the path to pyproject.toml
+    pyproject_path = Path(__file__).parent.parent.parent / "pyproject.toml"
+
+    # Read the version from pyproject.toml
+    with pyproject_path.open("rb") as f:
+        pyproject_data = tomllib.load(f)
+        return pyproject_data["project"]["version"]
+
+
+__version__ = load_version()

tests/test_init.py

@@ -0,0 +1,49 @@
+import importlib
+from unittest.mock import mock_open, patch
+
+import pytest
+
+
+def test_version_loading():
+    mock_toml_content = """
+    [project]
+    version = "1.0.0"
+    """
+
+    with patch("pathlib.Path.open", mock_open(read_data=mock_toml_content)):
+        with patch("tomllib.load") as mock_load:
+            mock_load.return_value = {"project": {"version": "1.0.0"}}
+
+            import djpress
+
+            importlib.reload(djpress)  # Reload the module to apply the mock
+
+            assert djpress.__version__ == "1.0.0"
+
+
+def test_version_loading_error():
+    with patch("pathlib.Path.open", mock_open()) as mock_file:
+        mock_file.side_effect = FileNotFoundError()
+
+        with pytest.raises(FileNotFoundError):
+            import djpress
+
+            importlib.reload(djpress)  # Reload the module to apply the mock
+
+
+def test_tomllib_import_error():
+    mock_toml_content = """
+    [project]
+    version = "1.0.0"
+    """
+
+    with patch.dict("sys.modules", {"tomllib": None}):
+        with patch("pathlib.Path.open", mock_open(read_data=mock_toml_content)):
+            with patch("tomli.load") as mock_load:
+                mock_load.return_value = {"project": {"version": "1.0.0"}}
+
+                import djpress
+
+                importlib.reload(djpress)  # Reload the module to apply the mock
+
+                assert djpress.__version__ == "1.0.0"