Function build_store

nodedb-vector/src/multivec/meta_embed.rs:87–105

Build a store with `n` documents, each with one unit-vector at dimension `i % dim`.

(n: u32, dim: usize, k: u8)

    /// Build a store with `n` documents, each with one unit-vector at
    /// dimension `i % dim`.
    fn build_store(n: u32, dim: usize, k: u8) -> MultiVectorStore {
        let mut store = MultiVectorStore::new(dim, MultiVecMode::MetaToken { k });
        for i in 0..n {
            let mut vecs: Vec<Vec<f32>> = Vec::new();
            for j in 0..k as usize {
                let mut v = vec![0.0f32; dim];
                // Each Meta Token of doc i points in direction (i + j) % dim.
                v[(i as usize + j) % dim] = 1.0;
                vecs.push(v);
            }
            store
                .insert(MultiVectorDoc {
                    doc_id: i,
                    vectors: vecs,
                })
                .unwrap();
        }
        store
    }

    #[test]
    fn search_returns_at_most_k_results() {