File tree Expand file tree Collapse file tree 1 file changed +6
-10
lines changed
Expand file tree Collapse file tree 1 file changed +6
-10
lines changed Original file line number Diff line number Diff line change @@ -12,22 +12,18 @@ def pytest_collection_modifyitems(config, items):
1212 run_benchmarks = marker_expr and (
1313 "benchmark" in marker_expr and "not benchmark" not in marker_expr
1414 )
15-
1615 if gas_benchmark_values :
1716 run_benchmarks = True
18-
1917 items_for_removal = []
2018 for i , item in enumerate (items ):
21- is_benchmark_test = Path (__file__ ).parent in Path (item .fspath ).parents
22-
19+ is_in_benchmark_dir = Path (__file__ ).parent in Path (item .fspath ).parents
20+ has_benchmark_marker = item .get_closest_marker ("benchmark" ) is not None
21+ is_benchmark_test = is_in_benchmark_dir or has_benchmark_marker
2322 if is_benchmark_test :
24- benchmark_marker = pytest .mark .benchmark
25-
26- if gas_benchmark_values :
27- gas_values = [int (v .strip ()) for v in gas_benchmark_values .split ("," )]
28- benchmark_marker = pytest .mark .benchmark (gas_values = gas_values )
23+ if is_in_benchmark_dir and not has_benchmark_marker :
24+ benchmark_marker = pytest .mark .benchmark
2925
30- item .add_marker (benchmark_marker )
26+ item .add_marker (benchmark_marker )
3127 if not run_benchmarks :
3228 items_for_removal .append (i )
3329
You can’t perform that action at this time.
0 commit comments