From eb016119a0e598e5469383a02948833458917830 Mon Sep 17 00:00:00 2001 From: Khaleel Al-Adhami Date: Wed, 12 Feb 2025 14:24:14 -0800 Subject: [PATCH] import templates beforehands --- tests/benchmarks/test_compilation.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/benchmarks/test_compilation.py b/tests/benchmarks/test_compilation.py index a7a10015e..dbea81023 100644 --- a/tests/benchmarks/test_compilation.py +++ b/tests/benchmarks/test_compilation.py @@ -4,11 +4,20 @@ from reflex.compiler.compiler import _compile_page, _compile_stateful_components from reflex.components.component import Component +def import_templates(): + # Importing the templates module to avoid the import time in the benchmark + import reflex.compiler.templates # noqa: F401 + + def test_compile_page(evaluated_page: Component, benchmark: BenchmarkFixture): + import_templates() + benchmark(lambda: _compile_page(evaluated_page, None)) def test_compile_stateful(evaluated_page: Component, benchmark: BenchmarkFixture): + import_templates() + benchmark(lambda: _compile_stateful_components([evaluated_page]))