disable numba for testing

mdtanker · Nov 16, 2023 · 473e087 · 473e087
1 parent ad4d6a1
commit 473e087
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 3 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -13,6 +13,7 @@ concurrency:
 
 env:
   FORCE_COLOR: 3
+  NUMBA_DISABLE_JIT: "1"
 
 jobs:
   pre-commit:

diff --git a/Makefile b/Makefile
@@ -19,7 +19,7 @@ test_coverage:
 	NUMBA_DISABLE_JIT=1 pytest
 
 test_numba:
-	NUMBA_DISABLE_JIT=0 pytest -rP -m use_numba1
+	NUMBA_DISABLE_JIT=0 pytest -rP -m use_numba
 
 format:
 	ruff format $(STYLE_CHECK_FILES)

diff --git a/noxfile.py b/noxfile.py
@@ -43,7 +43,22 @@ def tests(session: nox.Session) -> None:
     Run the unit and regular tests.
     """
     session.install(".[test]")
-    session.run("pytest", *session.posargs)
+
+    # run tests with numba jit disabled to get real coverage
+    session.run(
+        "pytest",
+        *session.posargs,
+        env={"NUMBA_DISABLE_JIT": "1"},
+    )
+
+    # run just the numba tests with numba jit enabled
+    session.run(
+        "pytest",
+        "-m",
+        "use_numba",
+        *session.posargs,
+        env={"NUMBA_DISABLE_JIT": "0"},
+    )
 
 
 @nox.session(reuse_venv=True)

diff --git a/pyproject.toml b/pyproject.toml
@@ -75,7 +75,9 @@ log_cli_level = "INFO"
 testpaths = [
   "tests",
 ]
-
+markers = [
+    "use_numba: mark test functions that call Numba jitted functions",
+]
 
 [tool.coverage]
 run.source = ["invert4geom"]