Add filter benchmark

Author: henadzit

tests/benchmarks/conftest.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 from decimal import Decimal
 import random
@@ -17,13 +19,13 @@ def setup_database():
 
 @pytest.fixture(scope="module", autouse=True)
 def skip_if_codspeed_not_enabled(request):
-    if not request.config.getoption("--codspeed"):
-        pytest.skip("codspeed tests are disabled")
+    if not request.config.getoption("--codspeed", default=None):
+        pytest.skip("codspeed is not enabled")
 
 
 @pytest.fixture
 def few_fields_benchmark_dataset() -> list[BenchmarkFewFields]:
-    async def _create():
+    async def _create() -> list[BenchmarkFewFields]:
         res = []
         for _ in range(100):
             res.append(await BenchmarkFewFields.create(level=random.randint(0, 100), text="test"))
@@ -34,7 +36,7 @@ async def _create():
 
 @pytest.fixture
 def many_fields_benchmark_dataset() -> list[BenchmarkManyFields]:
-    async def _create():
+    async def _create() -> list[BenchmarkManyFields]:
         res = []
         for _ in range(100):
             res.append(

tests/benchmarks/test_filter.py

@@ -0,0 +1,16 @@
+import asyncio
+import random
+
+from tests.testmodels import BenchmarkFewFields
+
+
+def test_filter_few_fields(benchmark, few_fields_benchmark_dataset):
+    loop = asyncio.get_event_loop()
+    levels = list(set([o.level for o in few_fields_benchmark_dataset]))
+
+    @benchmark
+    def bench():
+        async def _bench():
+            await BenchmarkFewFields.filter(level__in=random.sample(levels, 5)).limit(5)
+
+        loop.run_until_complete(_bench())