chore(test): add doctest namespace (#25)

* chore(test): add doctest namespace

Import JAX and NumPy in the doctest namespace to make examples less crowded by imports.

* chore(ci): apply lint suggestions

* fix(deps): conftest is now correctly removed

* fix(test): impossible to test conftest not present

* chore(ci): do not cover conftest

* chore(lint): apply fixes

* fix(ci): typo

* fix(ci): omit conftest

* chore(docs): remove trailing imports

docs/source/notebooks/plotting_backend.ipynb

@@ -182,7 +182,7 @@
    },
    "outputs": [],
    "source": [
-    "import differt.plotting as dplt\n",
+    "import differt.plotting as dplt  # noqa: E402\n",
     "\n",
     "dplt.use(\"plotly\")\n",
     "\n",

docs/source/notebooks/quickstart.ipynb

@@ -17,8 +17,8 @@
    "source": [
     "from pathlib import Path\n",
     "\n",
-    "import numpy as np\n",
     "import jax.numpy as jnp\n",
+    "import numpy as np\n",
     "\n",
     "import differt.plotting as dplt\n",
     "from differt.geometry import TriangleMesh\n",

pyproject.toml

@@ -27,7 +27,6 @@ dependencies = [
 ]
 description = "Differentiable Ray Tracing Toolbox for Radio Propagation Simulations"
 dynamic = ["license", "readme", "version"]
-exclude = ["docs/", "static/"]
 keywords = ["ray tracing", "differentiable", "propagation", "radio", "jax"]
 name = "DiffeRT"
 requires-python = ">=3.9"
@@ -58,8 +57,19 @@ exclude_lines = [
 ]
 precision = 2
 
+[tool.coverage.run]
+omit = ["**/*/conftest.py"]
+
 [tool.maturin]
 bindings = "pyo3"
+exclude = [
+  ".*",
+  ".github/**/*",
+  "benches/**/*",
+  "conftest.py",  # Path relative to source (python/differt/conftest.py does not work)
+  "docs/**/*",
+  "tests/**/*",
+]
 features = ["pyo3/extension-module"]
 module-name = "differt._core"
 python-source = "python"
@@ -138,7 +148,9 @@ target-version = "py39"
 extend-ignore-names = ["T"]
 
 [tool.ruff.lint.per-file-ignores]
+"**.ipynb" = ["B018"]
 "**/{tests,docs}/*" = ["D"]
+"python/differt/conftest.py" = ["D"]
 
 [tool.ruff.pydocstyle]
 convention = "google"

python/differt/conftest.py

@@ -0,0 +1,12 @@
+from typing import Any
+
+import jax
+import numpy
+import pytest
+
+
+@pytest.fixture(autouse=True)
+def add_doctest_modules(doctest_namespace: dict[str, Any]) -> None:
+    doctest_namespace["jax"] = jax
+    doctest_namespace["jnp"] = jax.numpy
+    doctest_namespace["np"] = numpy

python/differt/geometry/utils.py

@@ -37,7 +37,7 @@ def normalize(
     :Examples:
 
     >>> from differt.geometry.utils import normalize
-    >>> import jax.numpy as jnp
+    >>>
     >>> vector = jnp.array([1.0, 1.0, 1.0])
     >>> normalize(vector)  # [1., 1., 1.] / sqrt(3), sqrt(3)
     (Array([0.57735026, 0.57735026, 0.57735026], dtype=float32),

python/differt/rt/image_method.py

@@ -45,7 +45,6 @@ def image_of_vertices_with_respect_to_mirrors(
         a batch of random vertices. Here, normal vectors do not have a unit length,
         but they should have if you want an interpretable result.
 
-        >>> import jax
         >>> from differt.rt.image_method import (
         ...     image_of_vertices_with_respect_to_mirrors,
         ... )
@@ -137,8 +136,6 @@ def image_method(
 
         .. code-block:: python
 
-            import jax.numpy as jnp
-
             paths = image_method(
                 from_vertices, to_vertices, mirror_vertices, mirror_normals
             )

python/differt/utils.py

@@ -18,8 +18,6 @@ def sorted_array2(array: Shaped[Array, "m n"]) -> Shaped[Array, "m n"]:
     Examples:
         The following example shows how the sorting works.
 
-        >>> import jax
-        >>> import jax.numpy as jnp
         >>> from differt.utils import sorted_array2
         >>>
         >>> arr = jnp.arange(10).reshape(5, 2)