add stateful benchmarks

tests/benchmarks/conftest.py

@@ -1,3 +1,3 @@
-from .fixtures import evaluated_page
+from .fixtures import evaluated_page, unevaluated_page
 
-__all__ = ["evaluated_page"]
+__all__ = ["evaluated_page", "unevaluated_page"]

tests/benchmarks/fixtures.py

@@ -213,6 +213,33 @@ def side_bar():
     )
 
 
+class BenchmarkState(rx.State):
+    """State for the benchmark."""
+
+    counter: rx.Field[int] = rx.field(0)
+
+    @rx.event
+    def increment(self):
+        """Increment the counter."""
+        self.counter = self.counter + 1
+
+    @rx.event
+    def decrement(self):
+        """Decrement the counter."""
+        self.counter = self.counter - 1
+
+    @rx.var
+    def elements(self) -> list[int]:
+        """List of elements.
+
+        Returns:
+            List of elements.
+        """
+        if self.counter < 0:
+            return list(range(0))
+        return list(range(self.counter))
+
+
 LOREM_IPSUM = "Lorem ipsum dolor sit amet, dolor ut dolore pariatur aliqua enim tempor sed. Labore excepteur sed exercitation. Ullamco aliquip lorem sunt enim in incididunt. Magna anim officia sint cillum labore. Ut eu non dolore minim nostrud magna eu, aute ex in incididunt irure eu. Fugiat et magna magna est excepteur eiusmod minim. Quis eiusmod et non pariatur dolor veniam incididunt, eiusmod irure enim sed dolor lorem pariatur do. Occaecat duis irure excepteur dolore. Proident ut laborum pariatur sit sit, nisi nostrud voluptate magna commodo laborum esse velit. Voluptate non minim deserunt adipiscing irure deserunt cupidatat. Laboris veniam commodo incididunt veniam lorem occaecat, fugiat ipsum dolor cupidatat. Ea officia sed eu excepteur culpa adipiscing, tempor consectetur ullamco eu. Anim ex proident nulla sunt culpa, voluptate veniam proident est adipiscing sint elit velit. Laboris adipiscing est culpa cillum magna. Sit veniam nulla nulla, aliqua eiusmod commodo lorem cupidatat commodo occaecat. Fugiat cillum dolor incididunt mollit eiusmod sint. Non lorem dolore labore excepteur minim laborum sed. Irure nisi do lorem nulla sunt commodo, deserunt quis mollit consectetur minim et esse est, proident nostrud officia enim sed reprehenderit. Magna cillum consequat aute reprehenderit duis sunt ullamco. Labore qui mollit voluptate. Duis dolor sint aute amet aliquip officia, est non mollit tempor enim quis fugiat, eu do culpa consectetur magna. Do ullamco aliqua voluptate culpa excepteur reprehenderit reprehenderit. Occaecat nulla sit est magna. Deserunt ea voluptate veniam cillum. Amet cupidatat duis est tempor fugiat ex eu, officia est sunt consectetur labore esse exercitation. Nisi cupidatat irure est nisi. Officia amet eu veniam reprehenderit. In amet incididunt tempor commodo ea labore. Mollit dolor aliquip excepteur, voluptate aute occaecat id officia proident. Ullamco est amet tempor. Proident aliquip proident mollit do aliquip ipsum, culpa quis aute id irure. Velit excepteur cillum cillum ut cupidatat. Occaecat qui elit esse nulla minim. Consequat velit id ad pariatur tempor. Eiusmod deserunt aliqua ex sed quis non. Dolor sint commodo ex in deserunt nostrud excepteur, pariatur ex aliqua anim adipiscing amet proident. Laboris eu laborum magna lorem ipsum fugiat velit."
 
 
@@ -233,6 +260,27 @@ def _complicated_page():
     )
 
 
-@pytest.fixture(params=[_simple_page, _complicated_page])
+def _stateful_page():
+    return rx.hstack(
+        rx.text(BenchmarkState.counter),
+        rx.button("Increment", on_click=BenchmarkState.increment),
+        rx.button("Decrement", on_click=BenchmarkState.decrement),
+        rx.cond(
+            BenchmarkState.counter < 0,
+            rx.text("Counter is negative"),
+        ),
+        rx.foreach(
+            BenchmarkState.elements,
+            lambda elem: rx.text(elem),
+        ),
+    )
+
+
+@pytest.fixture(params=[_simple_page, _complicated_page, _stateful_page])
+def unevaluated_page(request):
+    return request.param
+
+
+@pytest.fixture(params=[_simple_page, _complicated_page, _stateful_page])
 def evaluated_page(request):
     return request.param()

tests/benchmarks/test_evaluate.py

@@ -1,9 +1,6 @@
 import pytest
 
-from .fixtures import _complicated_page, _simple_page
-
 
 @pytest.mark.benchmark
-@pytest.mark.parametrize("page", [_simple_page, _complicated_page])
-def test_evaluate_page(page):
-    page()
+def test_evaluate_page(unevaluated_page):
+    unevaluated_page()