Function prefetch_batch_then_fetch

nodedb/src/data/vamana_fetcher.rs:352–385

()

    #[test]
    fn prefetch_batch_then_fetch() {
        let dir = tempfile::tempdir().unwrap();
        let vecs: Vec<Vec<f32>> = (0..8u32)
            .map(|i| vec![i as f32, (i + 1) as f32, (i + 2) as f32, (i + 3) as f32])
            .collect();
        let dim = 4u32;
        let n = vecs.len() as u64;

        let layout = VamanaStorageLayout {
            dim,
            r: 0,
            alpha: 1.2,
            num_nodes: n,
            entry: 0,
            adjacency_offset: 0,
            vectors_offset: 0,
        };

        let path = dir.path().join("vamana.vec");
        write_vectors_file(&path, &vecs);

        let mut fetcher = IoUringNodeFetcher::open(&path, layout, 8).unwrap();

        // Pre-fetch nodes 0..4 in one batch.
        fetcher.prefetch_batch(&[0, 1, 2, 3]);

        for i in 0u32..4 {
            let got = fetcher.fetch_fp32(i).unwrap();
            assert_eq!(
                got, vecs[i as usize],
                "node {i} vector mismatch after prefetch"
            );
        }
    }

    #[test]
    fn out_of_range_returns_none() {