diff --git a/.github/workflows/performance.yml b/.github/workflows/performance.yml index c7bd1003a..cefa331de 100644 --- a/.github/workflows/performance.yml +++ b/.github/workflows/performance.yml @@ -31,4 +31,4 @@ jobs: uses: CodSpeedHQ/action@v3 with: token: ${{ secrets.CODSPEED_TOKEN }} - run: poetry run pytest benchmarks/test_evaluate.py --codspeed + run: poetry run pytest tests/benchmarks --codspeed diff --git a/pyproject.toml b/pyproject.toml index 2b5507a1d..b54f578fd 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -102,6 +102,7 @@ lint.pydocstyle.convention = "google" "*/blank.py" = ["I001"] [tool.pytest.ini_options] +filterwarnings = "ignore:fields may not start with an underscore:RuntimeWarning" asyncio_default_fixture_loop_scope = "function" asyncio_mode = "auto" diff --git a/tests/benchmarks/__init__.py b/tests/benchmarks/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/tests/benchmarks/conftest.py b/tests/benchmarks/conftest.py new file mode 100644 index 000000000..bf777be2e --- /dev/null +++ b/tests/benchmarks/conftest.py @@ -0,0 +1,3 @@ +from .fixtures import evaluated_page + +__all__ = ["evaluated_page"] diff --git a/benchmarks/test_evaluate.py b/tests/benchmarks/fixtures.py similarity index 97% rename from benchmarks/test_evaluate.py rename to tests/benchmarks/fixtures.py index aa4c8237e..d9c0d7688 100644 --- a/benchmarks/test_evaluate.py +++ b/tests/benchmarks/fixtures.py @@ -216,7 +216,14 @@ def side_bar(): 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." -def complicated_page(): +def _simple_page(): + return rx.box( + rx.heading("Simple Page", size="1"), + rx.text(LOREM_IPSUM), + ) + + +def _complicated_page(): return rx.hstack( side_bar(), rx.box( @@ -226,6 +233,6 @@ def complicated_page(): ) -@pytest.mark.benchmark -def test_component_init(): - complicated_page() +@pytest.fixture(params=[_simple_page, _complicated_page]) +def evaluated_page(request): + return request.param() diff --git a/tests/benchmarks/test_compilation.py b/tests/benchmarks/test_compilation.py new file mode 100644 index 000000000..0a20ed521 --- /dev/null +++ b/tests/benchmarks/test_compilation.py @@ -0,0 +1,18 @@ +import pytest + +from reflex.compiler.compiler import _compile_page, _compile_stateful_components + + +@pytest.mark.benchmark +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() diff --git a/tests/benchmarks/test_evaluate.py b/tests/benchmarks/test_evaluate.py new file mode 100644 index 000000000..c8a6b392d --- /dev/null +++ b/tests/benchmarks/test_evaluate.py @@ -0,0 +1,9 @@ +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()