mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-26 23:59:58 +00:00
30a7dd630c
## Problem `TYPE_CHECKING` is used inconsistently across Python tests. ## Summary of changes - Update `ruff`: 0.7.0 -> 0.11.2 - Enable TC (flake8-type-checking): https://docs.astral.sh/ruff/rules/#flake8-type-checking-tc - (auto)fix all new issues
31 lines
1.3 KiB
Python
31 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
|
|
from performance.test_perf_pgbench import PgBenchLoadType, get_durations_matrix, run_test_pgbench
|
|
|
|
if TYPE_CHECKING:
|
|
from fixtures.compare_fixtures import PgCompare
|
|
|
|
|
|
# The following test runs on an existing database that has pgvector extension installed
|
|
# and a table with 1 million embedding vectors loaded and indexed with HNSW.
|
|
#
|
|
# Run this pgbench tests against an existing remote Postgres cluster with the necessary setup.
|
|
@pytest.mark.parametrize("duration", get_durations_matrix())
|
|
@pytest.mark.remote_cluster
|
|
def test_pgbench_remote_pgvector_hnsw(remote_compare: PgCompare, duration: int):
|
|
run_test_pgbench(remote_compare, 1, duration, PgBenchLoadType.PGVECTOR_HNSW)
|
|
|
|
|
|
# The following test runs on an existing database that has pgvector extension installed
|
|
# and a table with 1 million embedding vectors loaded and indexed with halfvec.
|
|
#
|
|
# Run this pgbench tests against an existing remote Postgres cluster with the necessary setup.
|
|
@pytest.mark.parametrize("duration", get_durations_matrix())
|
|
@pytest.mark.remote_cluster
|
|
def test_pgbench_remote_pgvector_halfvec(remote_compare: PgCompare, duration: int):
|
|
run_test_pgbench(remote_compare, 1, duration, PgBenchLoadType.PGVECTOR_HALFVEC)
|