chore(ci): add bechmark for quads

jeertmans · Oct 10, 2024 · d1c8115 · d1c8115
1 parent f072acf
commit d1c8115
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/differt/tests/benchmarks/test_rt.py b/differt/tests/benchmarks/test_rt.py
@@ -61,13 +61,15 @@ def test_transmitter_visibility_in_simple_street_canyon_scene(
 @pytest.mark.benchmark(group="compute_paths")
 @pytest.mark.parametrize("order", [0, 1, 2])
 @pytest.mark.parametrize("chunk_size", [None, 20_000])
+@pytest.mark.parametrize("assume_quads", [False, True])
 def test_compute_paths_in_simple_street_canyon_scene(
     order: int,
     chunk_size: int | None,
+    assume_quads: bool,
     simple_street_canyon_scene: TriangleScene,
     benchmark: BenchmarkFixture,
 ) -> None:
-    scene = simple_street_canyon_scene
+    scene = simple_street_canyon_scene.set_assume_quads(assume_quads)
     if chunk_size:
 
         @jax.debug_nans(False)  # noqa: FBT003