From 0ae79b1b1830cbc84ac5b6ac7ba50a94e7ec9358 Mon Sep 17 00:00:00 2001 From: Khaleel Al-Adhami Date: Wed, 12 Feb 2025 13:59:23 -0800 Subject: [PATCH] do the same for test_evaluate_page --- tests/benchmarks/fixtures.py | 4 ++-- tests/benchmarks/test_evaluate.py | 13 +++++++++---- 2 files changed, 11 insertions(+), 6 deletions(-) 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)