Add code coverage

.pre-commit-config.yaml

@@ -11,3 +11,12 @@ repos:
       - id: trailing-whitespace
       - id: end-of-file-fixer
       - id: check-yaml
+
+  - repo: local
+    hooks:
+      - id: custom-script-py
+        name: Check Coverage
+        entry: poetry run python check_coverage.py
+        language: system
+        pass_filenames: false
+        args: [--fix]

check_coverage.py

@@ -0,0 +1,11 @@
+import subprocess
+
+
+def run_coverage():
+    try:
+        # Run tests with coverage
+        subprocess.run(["poetry", "run", "coverage", "run", "-m", "pytest"], check=True)
+        # Generate coverage report and check threshold
+        subprocess.run(["poetry", "run", "coverage", "report", "--fail-under=95"], check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Coverage check failed: {e}")

pyproject.toml

@@ -50,6 +50,11 @@ line-ending = "auto"
 docstring-code-format = true
 docstring-code-line-length = 100
 
+[tool.coverage.run]
+omit = [
+    "tests/*"
+]
+
 [tool.poetry.dependencies]
 python = "^3.9"
 numpy = "^1.21.2"
@@ -58,6 +63,10 @@ matplotlib = "3.9.0"
 requests = "^2.32.3"
 semantic-version = "^2.10.0"
 toml = "^0.10.2"
+coverage = "^7.6.1"
+pytest-cov = "^5.0.0"
+
+
 
 [tool.setuptools.packages.find]
 where = [ "src",]