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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
"""Compare benchmarks for AwesomeVersion."""
import pytest
from pytest_codspeed import BenchmarkFixture
from awesomeversion import AwesomeVersion
from .const import DEFAULT_RUNS
@pytest.mark.parametrize(
"input_a,operator, input_b",
(
pytest.param("9999", ">", "1", id="9999>1"),
pytest.param("9999", ">", "2020.1.1", id="9999>2020.1.1"),
pytest.param("1.2.3b6", ">", "1.2.3.dev4", id="1.2.3b6>1.2.3.dev4"),
pytest.param("1.2.3", "==", "1.2.3", id="1.2.3==1.2.3"),
pytest.param("1.2.3", "!=", "3.2.1", id="1.2.3!=3.2.1"),
),
)
def test_compare(
benchmark: BenchmarkFixture,
input_a: str,
operator: str,
input_b: str,
) -> None:
"""Benchmark for AwesomeVersion comparison."""
obj = AwesomeVersion(input_a)
if operator == ">":
@benchmark
def _run_benchmark() -> None:
for _ in range(DEFAULT_RUNS):
assert obj > input_b
elif operator == "<":
@benchmark
def _run_benchmark() -> None:
for _ in range(DEFAULT_RUNS):
assert obj < input_b
elif operator == "==":
@benchmark
def _run_benchmark() -> None:
for _ in range(DEFAULT_RUNS):
assert obj == input_b
elif operator == "!=":
@benchmark
def _run_benchmark() -> None:
for _ in range(DEFAULT_RUNS):
assert obj != input_b
|
