Skip to content

Commit a9631e5

Browse files
committed
perf: benchmarks for pint_eval
1 parent be61d6f commit a9631e5

File tree

2 files changed

+23
-0
lines changed

2 files changed

+23
-0
lines changed
+23
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
from __future__ import annotations
2+
3+
import pytest
4+
5+
from pint.pint_eval import plain_tokenizer, uncertainty_tokenizer
6+
7+
VALUES = [
8+
"1",
9+
"1 + 2 + 5",
10+
"10 m",
11+
"10 metros + 5 segundos",
12+
"10 metros * (5 segundos)",
13+
]
14+
15+
16+
def _tok(tok, value):
17+
return tuple(tok(value))
18+
19+
20+
@pytest.mark.parametrize("tokenizer", (plain_tokenizer, uncertainty_tokenizer))
21+
@pytest.mark.parametrize("value", VALUES)
22+
def test_pint_eval(benchmark, tokenizer, value):
23+
benchmark(_tok, tokenizer, value)

0 commit comments

Comments
 (0)