From ffb5b3eb5899eecc36c1a0902c66f18c58a6f864 Mon Sep 17 00:00:00 2001 From: William Patton Date: Fri, 10 May 2024 14:35:48 +0200 Subject: [PATCH] make pytest run the benchmarks --- benchmarks/benchmark_dep_graph.py | 6 ++---- pytest.ini | 3 +++ 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/benchmarks/benchmark_dep_graph.py b/benchmarks/benchmark_dep_graph.py index b134b927..694156bb 100644 --- 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 index eda845e5..c3289bcb 100644 --- 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_* \ No newline at end of file