Skip to content

Commit 9b01736

Browse files
committed
Benchmark for list conversations for retrieve conversation operation
1 parent 1eab660 commit 9b01736

1 file changed

Lines changed: 60 additions & 0 deletions

File tree

tests/benchmarks/test_app_database.py

Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1228,3 +1228,63 @@ def test_postgres_list_conversations_for_one_user_large_db(
12281228
None
12291229
"""
12301230
benchmark_list_conversations_for_one_user(benchmark, LARGE_DB_RECORDS_COUNT)
1231+
1232+
1233+
def test_postgres_retrieve_conversation_empty_db(
1234+
postgres_database: None, benchmark: BenchmarkFixture
1235+
) -> None:
1236+
"""Benchmark retrieving conversations on an empty database.
1237+
1238+
Parameters:
1239+
postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1240+
benchmark (BenchmarkFixture): pytest-benchmark fixture.
1241+
1242+
Returns:
1243+
None
1244+
"""
1245+
benchmark_retrieve_conversation(benchmark, 0)
1246+
1247+
1248+
def test_postgres_retrieve_conversation_small_db(
1249+
postgres_database: None, benchmark: BenchmarkFixture
1250+
) -> None:
1251+
"""Benchmark retrieving conversations on a small database.
1252+
1253+
Parameters:
1254+
postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1255+
benchmark (BenchmarkFixture): pytest-benchmark fixture.
1256+
1257+
Returns:
1258+
None
1259+
"""
1260+
benchmark_retrieve_conversation(benchmark, SMALL_DB_RECORDS_COUNT)
1261+
1262+
1263+
def test_postgres_retrieve_conversation_middle_db(
1264+
postgres_database: None, benchmark: BenchmarkFixture
1265+
) -> None:
1266+
"""Benchmark retrieving conversations on a medium-sized database.
1267+
1268+
Parameters:
1269+
postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1270+
benchmark (BenchmarkFixture): pytest-benchmark fixture.
1271+
1272+
Returns:
1273+
None
1274+
"""
1275+
benchmark_retrieve_conversation(benchmark, MIDDLE_DB_RECORDS_COUNT)
1276+
1277+
1278+
def test_postgres_retrieve_conversation_large_db(
1279+
postgres_database: None, benchmark: BenchmarkFixture
1280+
) -> None:
1281+
"""Benchmark retrieving conversations on a large database.
1282+
1283+
Parameters:
1284+
postgres_database: Fixture that prepares a temporary PostgreSQL DB.
1285+
benchmark (BenchmarkFixture): pytest-benchmark fixture.
1286+
1287+
Returns:
1288+
None
1289+
"""
1290+
benchmark_retrieve_conversation(benchmark, LARGE_DB_RECORDS_COUNT)

0 commit comments

Comments
 (0)