Update tests

This commit is contained in:
Mubelotix
2025-07-25 13:41:31 +02:00
parent d243504296
commit a439f57d70
4 changed files with 34 additions and 12 deletions

View File

@ -201,10 +201,6 @@ async fn simple_search() {
} }
], ],
"query": "Captain", "query": "Captain",
"queryVector": [
1.0,
1.0
],
"processingTimeMs": "[duration]", "processingTimeMs": "[duration]",
"limit": 20, "limit": 20,
"offset": 0, "offset": 0,

View File

@ -3703,7 +3703,7 @@ async fn federation_vector_two_indexes() {
]})) ]}))
.await; .await;
snapshot!(code, @"200 OK"); snapshot!(code, @"200 OK");
snapshot!(json_string!(response, { ".processingTimeMs" => "[duration]", ".**._rankingScore" => "[score]" }), @r###" snapshot!(json_string!(response, { ".processingTimeMs" => "[duration]", ".**._rankingScore" => "[score]" }), @r#"
{ {
"hits": [ "hits": [
{ {
@ -3911,9 +3911,20 @@ async fn federation_vector_two_indexes() {
"limit": 20, "limit": 20,
"offset": 0, "offset": 0,
"estimatedTotalHits": 8, "estimatedTotalHits": 8,
"queryVectors": {
"0": [
1.0,
0.0,
0.5
],
"1": [
0.8,
0.6
]
},
"semanticHitCount": 6 "semanticHitCount": 6
} }
"###); "#);
// hybrid search, distinct embedder // hybrid search, distinct embedder
let (response, code) = server let (response, code) = server

View File

@ -323,15 +323,20 @@ async fn binary_quantize_clear_documents() {
// Make sure the arroy DB has been cleared // Make sure the arroy DB has been cleared
let (documents, _code) = let (documents, _code) =
index.search_post(json!({ "hybrid": { "embedder": "manual" }, "vector": [1, 1, 1] })).await; index.search_post(json!({ "hybrid": { "embedder": "manual" }, "vector": [1, 1, 1] })).await;
snapshot!(documents, @r###" snapshot!(documents, @r#"
{ {
"hits": [], "hits": [],
"query": "", "query": "",
"queryVector": [
1.0,
1.0,
1.0
],
"processingTimeMs": "[duration]", "processingTimeMs": "[duration]",
"limit": 20, "limit": 20,
"offset": 0, "offset": 0,
"estimatedTotalHits": 0, "estimatedTotalHits": 0,
"semanticHitCount": 0 "semanticHitCount": 0
} }
"###); "#);
} }

View File

@ -685,17 +685,22 @@ async fn clear_documents() {
// Make sure the arroy DB has been cleared // Make sure the arroy DB has been cleared
let (documents, _code) = let (documents, _code) =
index.search_post(json!({ "vector": [1, 1, 1], "hybrid": {"embedder": "manual"} })).await; index.search_post(json!({ "vector": [1, 1, 1], "hybrid": {"embedder": "manual"} })).await;
snapshot!(documents, @r###" snapshot!(documents, @r#"
{ {
"hits": [], "hits": [],
"query": "", "query": "",
"queryVector": [
1.0,
1.0,
1.0
],
"processingTimeMs": "[duration]", "processingTimeMs": "[duration]",
"limit": 20, "limit": 20,
"offset": 0, "offset": 0,
"estimatedTotalHits": 0, "estimatedTotalHits": 0,
"semanticHitCount": 0 "semanticHitCount": 0
} }
"###); "#);
} }
#[actix_rt::test] #[actix_rt::test]
@ -739,7 +744,7 @@ async fn add_remove_one_vector_4588() {
json!({"vector": [1, 1, 1], "hybrid": {"semanticRatio": 1.0, "embedder": "manual"} }), json!({"vector": [1, 1, 1], "hybrid": {"semanticRatio": 1.0, "embedder": "manual"} }),
) )
.await; .await;
snapshot!(documents, @r###" snapshot!(documents, @r#"
{ {
"hits": [ "hits": [
{ {
@ -748,13 +753,18 @@ async fn add_remove_one_vector_4588() {
} }
], ],
"query": "", "query": "",
"queryVector": [
1.0,
1.0,
1.0
],
"processingTimeMs": "[duration]", "processingTimeMs": "[duration]",
"limit": 20, "limit": 20,
"offset": 0, "offset": 0,
"estimatedTotalHits": 1, "estimatedTotalHits": 1,
"semanticHitCount": 1 "semanticHitCount": 1
} }
"###); "#);
let (documents, _code) = index let (documents, _code) = index
.get_all_documents(GetAllDocumentsOptions { retrieve_vectors: true, ..Default::default() }) .get_all_documents(GetAllDocumentsOptions { retrieve_vectors: true, ..Default::default() })