Function test_usv_index_node_serialization

src/models/serializer/usv/tests.rs:121–169

()

#[test]
fn test_usv_index_node_serialization() {
    let quantization_bits = 6;
    let mut rng = thread_rng();
    let (dim_bufman, data_bufmans, cache, cursor, _temp_dir) = setup_test(quantization_bits);

    let data = USVIndexNode::new(0, quantization_bits, FileOffset(0));

    cache
        .offset_counter
        .fetch_add(USVIndexNode::get_serialized_size(), Ordering::Relaxed);

    for vector_id in 0..100 {
        let len = rng.gen_range(10..20);
        for _ in 0..len {
            let quotient = rng.gen_range(100..2000);
            let value = rng.gen_range(0.0..1.0);
            data.insert(
                quotient,
                value,
                vector_id,
                &cache,
                VersionNumber::from(0),
                1.0,
            )
            .unwrap();
        }
    }

    let offset = data
        .serialize(
            &dim_bufman,
            &data_bufmans,
            quantization_bits,
            &cache.offset_counter,
            cursor,
        )
        .unwrap();

    let deserialized = USVIndexNode::deserialize(
        &dim_bufman,
        &data_bufmans,
        FileOffset(offset),
        VersionNumber::from(u32::MAX),
        &cache,
    )
    .unwrap();

    assert_eq!(data, deserialized);
}

#[test]
fn end_to_end_test() {