make pytest run the benchmarks

funkelab · May 10, 2024 · ffb5b3e · ffb5b3e
1 parent 1aeae0a
commit ffb5b3e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/benchmarks/benchmark_dep_graph.py b/benchmarks/benchmark_dep_graph.py
@@ -46,14 +46,12 @@
 
 @pytest.mark.parametrize("test", ["iterate", "init"])
 @pytest.mark.parametrize("size", ["small", "medium", "large"])
-@pytest.mark.parametrize("block_gen", ["lazy", "enumerated"])
-def benchmark_dep_graph(benchmark, test, size, block_gen):
+def benchmark_dep_graph(benchmark, test, size):
     task = tasks[size]
-    lazy = block_gen == "lazy"
     iterate = test == "iterate"
 
     def benchmark_scheduler(task, block_count):
-        scheduler = Scheduler([task], lazy=lazy)
+        scheduler = Scheduler([task])
         for i in range(block_count):
             block = scheduler.acquire_block(task.task_id)
             assert block is not None, f"Failed to get the {i}'th block!"

diff --git a/pytest.ini b/pytest.ini
@@ -4,3 +4,6 @@ filterwarnings =
   ignore:::zarr
 log_cli=false
 log_level=DEBUG
+testpaths = tests benchmarks
+python_files = test_*.py benchmark_*.py
+python_functions = test_* benchmark_*