From 1b9671b6cc1d83e2a9c10ec362d57cfc96e74af9 Mon Sep 17 00:00:00 2001 From: Lendemor Date: Wed, 5 Feb 2025 19:39:32 +0100 Subject: [PATCH] add tests for _compile_stateful_components --- tests/benchmarks/test_compilation.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/benchmarks/test_compilation.py b/tests/benchmarks/test_compilation.py index e9640d4e3..0a20ed521 100644 --- a/tests/benchmarks/test_compilation.py +++ b/tests/benchmarks/test_compilation.py @@ -1,6 +1,6 @@ import pytest -from reflex.compiler.compiler import _compile_page +from reflex.compiler.compiler import _compile_page, _compile_stateful_components @pytest.mark.benchmark @@ -8,6 +8,11 @@ def test_compile_page(evaluated_page): _compile_page(evaluated_page, None) +@pytest.mark.benchmark +def test_compile_stateful(evaluated_page): + _compile_stateful_components([evaluated_page]) + + @pytest.mark.benchmark def test_get_all_imports(evaluated_page): evaluated_page._get_all_imports()