diff --git a/tests/benchmarks/conftest.py b/tests/benchmarks/conftest.py index f80093124..bf777be2e 100644 --- a/tests/benchmarks/conftest.py +++ b/tests/benchmarks/conftest.py @@ -1,9 +1,3 @@ -import warnings +from .fixtures import evaluated_page -from .fixtures import complicated_page - -warnings.filterwarnings( - "ignore", message="fields may not start with an underscore", category=RuntimeWarning -) - -__all__ = ["complicated_page"] +__all__ = ["evaluated_page"] diff --git a/tests/benchmarks/fixtures.py b/tests/benchmarks/fixtures.py index 7aa7f9e2a..d9c0d7688 100644 --- a/tests/benchmarks/fixtures.py +++ b/tests/benchmarks/fixtures.py @@ -234,5 +234,5 @@ def _complicated_page(): @pytest.fixture(params=[_simple_page, _complicated_page]) -def complicated_page(request): +def evaluated_page(request): return request.param() diff --git a/tests/benchmarks/test_compilation.py b/tests/benchmarks/test_compilation.py index 0578f1ef5..e9640d4e3 100644 --- a/tests/benchmarks/test_compilation.py +++ b/tests/benchmarks/test_compilation.py @@ -4,10 +4,10 @@ from reflex.compiler.compiler import _compile_page @pytest.mark.benchmark -def test_compile_page(complicated_page): - _compile_page(complicated_page, None) +def test_compile_page(evaluated_page): + _compile_page(evaluated_page, None) @pytest.mark.benchmark -def test_get_all_imports(complicated_page): - complicated_page._get_all_imports() +def test_get_all_imports(evaluated_page): + evaluated_page._get_all_imports()