blob: 67097a34e8547c708f7ece6565a1762173d82852 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
"""Benchmark: match hot path across matches, near-matches, and non-matches."""
from __future__ import annotations
import pytest
from pytest_benchmark.fixture import BenchmarkFixture
from edify.result import Regex
from tests.bench.cases import BUILDER_FACTORIES, MATCH_SCENARIOS
_MATCH_KINDS = ("matches", "near_matches", "non_matches")
@pytest.mark.parametrize("case_name", list(BUILDER_FACTORIES))
@pytest.mark.parametrize("match_kind", _MATCH_KINDS)
def test_match_bench(case_name: str, match_kind: str, benchmark: BenchmarkFixture):
factory = BUILDER_FACTORIES[case_name]
compiled = factory().to_regex()
inputs = MATCH_SCENARIOS[case_name][match_kind]
benchmark(_match_all_inputs, compiled, inputs)
def _match_all_inputs(compiled: Regex, inputs: list[str]) -> None:
for candidate in inputs:
compiled.search(candidate)
|
