turbo-quant 0.2.0

Experimental vector compression sidecars with PolarQuant, TurboQuant, QJL sketches, wire formats, and benchmark receipts
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

use turbo_quant::{
    CodecProfileV1, CompressionReceiptV1, PolarCode, QjlSketch, TurboCode, TurboQuantizer,
};

#[test]
fn legacy_codes_serialize_logical_fields() {
    let q = TurboQuantizer::new(16, 8, 8, 42).unwrap();
    let code = q.encode(&[0.1; 16]).unwrap();
    let json = serde_json::to_string(&code).unwrap();
    assert!(json.contains("angle_indices"));
    assert!(json.contains("signs"));
    let restored: TurboCode = serde_json::from_str(&json).unwrap();
    assert_eq!(code, restored);
}

#[test]
fn profile_and_receipt_serialize_roundtrip() {
    let q = TurboQuantizer::new(16, 8, 8, 42).unwrap();
    let (_code, receipt) = q
        .encode_with_receipt(&[0.1; 16], Some("source:test".into()))
        .unwrap();
    let json = serde_json::to_string(&receipt).unwrap();
    let restored: CompressionReceiptV1 = serde_json::from_str(&json).unwrap();
    let profile: CodecProfileV1 =
        serde_json::from_value(serde_json::to_value(&restored.profile).unwrap()).unwrap();
    assert_eq!(restored, receipt);
    assert_eq!(profile.profile_digest, q.profile().profile_digest);
}

#[test]
fn direct_packed_structs_validate_after_roundtrip() {
    let polar = PolarCode {
        dim: 8,
        bits: 4,
        radii: vec![1.0; 4],
        angle_indices: vec![1, 2, 3, 4],
    };
    let qjl = QjlSketch {
        dim: 8,
        projections: 8,
        signs: vec![1, -1, 1, -1, 1, -1, 1, -1],
    };
    polar.validate_for(8, 4).unwrap();
    qjl.validate_for(8, 8).unwrap();
}