velesdb-core 1.11.0

High-performance vector database engine written in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

#![allow(
    clippy::cast_precision_loss,
    clippy::cast_possible_truncation,
    clippy::cast_sign_loss,
    clippy::float_cmp
)]
//! Tests for `async_ops` module - Async wrappers for collection operations.

use super::async_ops::*;
use crate::collection::types::Collection;
use crate::point::Point;
use crate::DistanceMetric;
use std::sync::Arc;
use tempfile::TempDir;

fn create_test_collection() -> (TempDir, Arc<Collection>) {
    let dir = TempDir::new().unwrap();
    let collection =
        Collection::create(dir.path().to_path_buf(), 4, DistanceMetric::Cosine).unwrap();
    (dir, Arc::new(collection))
}

#[tokio::test]
async fn test_upsert_bulk_async() {
    let (_dir, collection) = create_test_collection();

    let points: Vec<Point> = (0..100)
        .map(|i| Point::without_payload(i, vec![i as f32; 4]))
        .collect();

    let result = upsert_bulk_async(collection, points).await;
    assert!(result.is_ok());
    assert_eq!(result.unwrap(), 100);
}

#[tokio::test]
async fn test_upsert_bulk_streaming() {
    let (_dir, collection) = create_test_collection();

    let points: Vec<Point> = (0..500)
        .map(|i| Point::without_payload(i, vec![i as f32; 4]))
        .collect();

    let result = upsert_bulk_streaming(collection, points, 100, None::<fn(usize, usize)>).await;
    assert!(result.is_ok());
    assert_eq!(result.unwrap(), 500);
}

#[tokio::test]
async fn test_search_async() {
    let (_dir, collection) = create_test_collection();

    let points: Vec<Point> = (0..10)
        .map(|i| Point::without_payload(i, vec![i as f32; 4]))
        .collect();
    collection.upsert(points).unwrap();

    let query = vec![5.0f32; 4];
    let result = search_async(collection, query, 3).await;
    assert!(result.is_ok());
    assert_eq!(result.unwrap().len(), 3);
}

#[tokio::test]
async fn test_flush_async() {
    let (_dir, collection) = create_test_collection();

    let result = flush_async(collection).await;
    assert!(result.is_ok());
}