Function build_store

nodedb-vector/src/multivec/plaid.rs:297–329

()

    use crate::multivec::storage::{MultiVecMode, MultiVectorDoc, MultiVectorStore};

    fn build_store() -> MultiVectorStore {
        let mut store = MultiVectorStore::new(2, MultiVecMode::PerToken);

        // Three well-separated clusters of documents.
        // Cluster A: docs 0–2 near (0,0).
        // Cluster B: docs 3–5 near (10,0).
        // Cluster C: docs 6–8 near (0,10).
        for i in 0u32..3 {
            store
                .insert(MultiVectorDoc {
                    doc_id: i,
                    vectors: vec![vec![i as f32 * 0.1, i as f32 * 0.1]],
                })
                .unwrap();
        }
        for i in 3u32..6 {
            store
                .insert(MultiVectorDoc {
                    doc_id: i,
                    vectors: vec![vec![10.0 + i as f32 * 0.1, 0.0]],
                })
                .unwrap();
        }
        for i in 6u32..9 {
            store
                .insert(MultiVectorDoc {
                    doc_id: i,
                    vectors: vec![vec![0.0, 10.0 + i as f32 * 0.1]],
                })
                .unwrap();
        }
        store
    }

    #[test]
    fn train_produces_correct_centroid_count() {