sqlite-vector-rs 0.2.2

SQLite extension providing PGVector-like native vector types with HNSW indexing
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

use sqlite_vector_rs::json::{blob_to_json, json_to_blob};
use sqlite_vector_rs::types::VectorType;

#[test]
fn json_to_float4_blob() {
    let blob = json_to_blob("[1.0, 2.0, 3.0]", VectorType::Float4).unwrap();
    let values: &[f32] = VectorType::Float4.blob_to_slice(&blob);
    assert_eq!(values, &[1.0, 2.0, 3.0]);
}

#[test]
fn json_to_float2_blob() {
    let blob = json_to_blob("[1.0, 2.0]", VectorType::Float2).unwrap();
    assert_eq!(blob.len(), 4); // 2 elements * 2 bytes
}

#[test]
fn json_to_int1_blob() {
    let blob = json_to_blob("[1, 2, -3]", VectorType::Int1).unwrap();
    let values: &[i8] = VectorType::Int1.blob_to_slice(&blob);
    assert_eq!(values, &[1, 2, -3]);
}

#[test]
fn blob_to_json_float4() {
    let values: Vec<f32> = vec![1.5, 2.5, 3.5];
    let blob = VectorType::Float4.slice_to_blob(&values);
    let json = blob_to_json(&blob, VectorType::Float4).unwrap();
    assert_eq!(json, "[1.5,2.5,3.5]");
}

#[test]
fn json_rejects_non_array() {
    assert!(json_to_blob("{}", VectorType::Float4).is_err());
    assert!(json_to_blob("42", VectorType::Float4).is_err());
}

#[test]
fn json_rejects_non_numeric_elements() {
    assert!(json_to_blob("[1.0, \"hello\", 3.0]", VectorType::Float4).is_err());
}

#[test]
fn json_rejects_nan() {
    assert!(json_to_blob("[1.0, null, 3.0]", VectorType::Float4).is_err());
}

#[test]
fn json_empty_array() {
    let blob = json_to_blob("[]", VectorType::Float4).unwrap();
    assert!(blob.is_empty());
}