add benchmarks for get_for_dialect (#1862)

Author: markus-96

CHANGELOG.rst

@@ -6,6 +6,18 @@ Changelog
 
 .. rst-class:: emphasize-children
 
+0.25
+====
+
+0.25.0 (unreleased)
+------
+Fixed
+^^^^^
+
+Changed
+^^^^^^^
+- add benchmarks for `get_for_dialect` (#1862)
+
 0.24
 ====
 

tests/benchmarks/conftest.py

@@ -6,7 +6,14 @@
 
 import pytest
 
-from tests.testmodels import BenchmarkFewFields, BenchmarkManyFields
+from tests.testmodels import (
+    BenchmarkFewFields,
+    BenchmarkManyFields,
+    Tournament,
+    Event,
+    Team,
+    DecimalFields,
+)
 from tortoise.contrib.test import _restore_default, truncate_all_models
 
 
@@ -87,3 +94,22 @@ def _gen():
         }
 
     return _gen
+
+
+@pytest.fixture
+def create_team_with_participants() -> None:
+    async def _create() -> None:
+        tournament = await Tournament.create(name="New Tournament")
+        event = await Event.create(name="Test", tournament_id=tournament.id)
+        team = await Team.create(name="Some Team")
+        await event.participants.add(team)
+
+    asyncio.get_event_loop().run_until_complete(_create())
+
+
+@pytest.fixture
+def create_decimals() -> None:
+    async def _create() -> None:
+        await DecimalFields.create(decimal=Decimal("1.23456"), decimal_nodec=18.7)
+
+    asyncio.get_event_loop().run_until_complete(_create())

tests/benchmarks/test_expressions.py

@@ -0,0 +1,27 @@
+import asyncio
+
+from tests.testmodels import BenchmarkFewFields, DecimalFields
+from tortoise.expressions import F
+from tortoise.functions import Count
+
+
+def test_expressions_count(benchmark, few_fields_benchmark_dataset):
+    loop = asyncio.get_event_loop()
+
+    @benchmark
+    def bench():
+        async def _bench():
+            await BenchmarkFewFields.annotate(text_count=Count("text"))
+
+        loop.run_until_complete(_bench())
+
+
+def test_expressions_f(benchmark, create_decimals):
+    loop = asyncio.get_event_loop()
+
+    @benchmark
+    def bench():
+        async def _bench():
+            await DecimalFields.annotate(d=F("decimal")).all()
+
+        loop.run_until_complete(_bench())

tests/benchmarks/test_field_attribute_lookup.py

@@ -0,0 +1,32 @@
+from tortoise.fields import Field
+
+
+class MyField(Field):
+    @property
+    def MY_PROPERTY(self):
+        return f"hi from {self.__class__.__name__}!"
+
+    OTHER_PROPERTY = "something else"
+
+    class _db_property:
+        def __init__(self, field: "Field"):
+            self.field = field
+
+        @property
+        def MY_PROPERTY(self):
+            return f"hi from {self.__class__.__name__} of {self.field.__class__.__name__}!"
+
+    class _db_cls_attribute:
+        MY_PROPERTY = "cls_attribute"
+
+
+def test_field_attribute_lookup_get_for_dialect(benchmark):
+    field = MyField()
+
+    @benchmark
+    def bench():
+        field.get_for_dialect("property", "MY_PROPERTY")
+        field.get_for_dialect("postgres", "MY_PROPERTY")
+        field.get_for_dialect("cls_attribute", "MY_PROPERTY")
+        field.get_for_dialect("property", "OTHER_PROPERTY")
+        field.get_for_dialect("property", "MY_PROPERTY")

tests/benchmarks/test_relations.py

@@ -0,0 +1,14 @@
+import asyncio
+
+from tests.testmodels import Event
+
+
+def test_relations_values_related_m2m(benchmark, create_team_with_participants):
+    loop = asyncio.get_event_loop()
+
+    @benchmark
+    def bench():
+        async def _bench():
+            await Event.all().values("participants__name")
+
+        loop.run_until_complete(_bench())