xiaoxue_admin
/
zvec
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ut: fix some uts (#394)

This commit is contained in:
egolearner 2026-05-12 19:04:33 +08:00 committed by GitHub
parent d91867e00e
commit d0489da401
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
python/tests/test_hnsw_contiguous_memory.py
View File

@ -46,7 +46,7 @@ from zvec import (
HnswIndexParam,
HnswQueryParam,
InvertIndexParam,
VectorQuery,
Query,
VectorSchema,
)
from zvec.typing import DataType, IndexType, MetricType, QuantizeType
@ -115,12 +115,12 @@ def _generate_docs(rng: np.random.Generator, num: int = NUM_DOCS) -> list[Doc]:
def _assert_query_matches(coll: Collection, query_vec: list[float]) -> list[str]:
"""Run a top-k vector query and return the returned ids in order."""
vector_query = VectorQuery(
vector_query = Query(
field_name="dense",
vector=query_vec,
param=HnswQueryParam(ef=128),
)
hits = coll.query(vectors=vector_query, topk=TOPK)
hits = coll.query(vector_query, topk=TOPK)
# Expect a single result group for the single vector query.
assert hits is not None, "query returned None"
assert len(hits) >= 1, f"expected at least one hit, got {hits!r}"

6
python/tests/test_vamana.py
View File

@ -46,7 +46,7 @@ from zvec import (
InvertIndexParam,
VamanaIndexParam,
VamanaQueryParam,
VectorQuery,
Query,
VectorSchema,
)
from zvec.typing import DataType, IndexType, MetricType, QuantizeType
@ -125,12 +125,12 @@ def _query_topk(
coll: Collection, query_vec: list[float], *, ef_search: int = 64
) -> list[str]:
"""Run a top-k vector query and return the returned ids in order."""
vector_query = VectorQuery(
vector_query = Query(
field_name="dense",
vector=query_vec,
param=VamanaQueryParam(ef_search=ef_search),
)
hits = coll.query(vectors=vector_query, topk=TOPK)
hits = coll.query(vector_query, topk=TOPK)
assert hits is not None, "query returned None"
assert len(hits) >= 1, f"expected at least one hit, got {hits!r}"
return [doc.id for doc in hits]