Coverage for tests / unit / benchmark_parallel / test_benchmark_demo.py: 100%

1"""Simple demo of using the benchmark script.""" 

2 

3from pathlib import Path 

4 

5from .benchmark_parallel import io_bound_task, light_cpu_task, main 

6 

7 

8def test_main(): 

9 """Test the main function of the benchmark script.""" 

10 _ = main( 

11 data_sizes=(1, 2), 

12 base_path=Path("tests/_temp/benchmark_demo"), 

13 plot=True, 

14 task_funcs={ 

15 "io_bound": io_bound_task, 

16 "light_cpu": light_cpu_task, 

17 }, 

18 )