rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-10-28 14:36:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(rust): maintain vector index in RTE geo test

Browse Source
This commit is contained in:
Xuanwo
2026-10-27 09:20:47 +08:00
parent a001db6b66
commit af1535fee7
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rust/lancedb/src/query.rs
+11 -1
View File
@@ -3361,11 +3361,21 @@ mod tests {
#[cfg(feature = "geo")]
#[tokio::test]
async fn test_st_dwithin_filter_on_rte_base_and_active_memtable() {
use crate::index::vector::IvfPqIndexBuilder;
let tmp_dir = tempdir().unwrap();
let table = make_geo_query_table(&tmp_dir).await;
table.set_unenforced_primary_key(["id"]).await.unwrap();
table
.enable_rte(crate::table::RTEWriteSpec::unsharded())
.create_index(&["vector"], Index::IvfPq(IvfPqIndexBuilder::default()))
.execute()
.await
.unwrap();
let vec_index = table.list_indices().await.unwrap()[0].name.clone();
table
.enable_rte(
crate::table::RTEWriteSpec::unsharded().with_maintained_indexes(vec![vec_index]),
)
.await
.unwrap();