diff --git a/tests/benchmarks/fixtures.py b/tests/benchmarks/fixtures.py index 16233100a..334d48282 100644 --- a/tests/benchmarks/fixtures.py +++ b/tests/benchmarks/fixtures.py @@ -374,10 +374,10 @@ def _stateful_page(): @pytest.fixture(params=[_simple_page, _complicated_page, _stateful_page]) -def unevaluated_page(request): +def unevaluated_page(request: pytest.FixtureRequest): return request.param @pytest.fixture(params=[_simple_page, _complicated_page, _stateful_page]) -def evaluated_page(request): +def evaluated_page(request: pytest.FixtureRequest): return request.param() diff --git a/tests/benchmarks/test_evaluate.py b/tests/benchmarks/test_evaluate.py index fbc75dea7..7fd75fc6c 100644 --- a/tests/benchmarks/test_evaluate.py +++ b/tests/benchmarks/test_evaluate.py @@ -1,6 +1,11 @@ -import pytest +from typing import Callable + +from pytest_codspeed import BenchmarkFixture + +from reflex.components.component import Component -@pytest.mark.benchmark -def test_evaluate_page(unevaluated_page): - unevaluated_page() +def test_evaluate_page( + unevaluated_page: Callable[[], Component], benchmark: BenchmarkFixture +): + benchmark(unevaluated_page)