import templates beforehands

reflex-dev · Feb 12, 2025 · eb01611 · eb01611
1 parent 0ae79b1
commit eb01611
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/benchmarks/test_compilation.py b/tests/benchmarks/test_compilation.py
@@ -4,11 +4,20 @@
 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]))