Function fetch_single_node_no_prefetch

nodedb/src/data/vamana_fetcher.rs:318–349

()

    #[test]
    fn fetch_single_node_no_prefetch() {
        let dir = tempfile::tempdir().unwrap();
        let vecs = vec![
            vec![1.0_f32, 2.0, 3.0, 4.0],
            vec![5.0_f32, 6.0, 7.0, 8.0],
            vec![9.0_f32, 10.0, 11.0, 12.0],
        ];
        let dim = 4u32;
        let n = vecs.len() as u64;

        // Build a layout with adjacency_offset = HEADER_BYTES = 64, r = 0.
        // vectors_offset = 64 + 0 * 4 * 3 = 64.
        // We write the vectors block starting at byte 0 of our test file,
        // so we create a layout where vectors_offset = 0.
        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();

        let got = fetcher.fetch_fp32(1).unwrap();
        assert_eq!(got, vecs[1], "node 1 vector mismatch");
    }

    #[test]
    fn prefetch_batch_then_fetch() {