audiofp 0.3.7

Pure-Rust audio fingerprinting and identification: Wang, Panako, Haitsma–Kalker, ONNX neural embedder, AudioSeal watermark, and streaming variants. no_std + alloc capable, bytemuck-friendly hash types.
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

//! Public-API integration tests for the watermark detector.
//!
//! These exercise the public contract: error handling, configuration
//! validation, and result types. End-to-end inference is not tested here
//! because it requires a shipped ONNX model; in-tree unit tests cover
//! Tract wiring via corrupt-file validation.

#![cfg(feature = "watermark")]

use audiofp::AfpError;
use audiofp::watermark::{WatermarkConfig, WatermarkDetector};

#[test]
fn config_new_uses_audioseal_defaults() {
    let cfg = WatermarkConfig::new("model.onnx");
    assert_eq!(cfg.message_bits, 16);
    assert_eq!(cfg.threshold, 0.5);
    assert_eq!(cfg.sample_rate, 16_000);
    assert_eq!(cfg.model_path, "model.onnx");
}

#[test]
fn empty_model_path_returns_model_not_found() {
    let res = WatermarkDetector::new(WatermarkConfig::new(""));
    match res {
        Err(AfpError::ModelNotFound(path)) => assert!(path.is_empty()),
        Ok(_) => panic!("expected ModelNotFound, got Ok"),
        Err(e) => panic!("expected ModelNotFound, got Err({e:?})"),
    }
}

#[test]
fn missing_model_file_returns_model_not_found() {
    let cfg = WatermarkConfig::new("/nonexistent/path/to/audioseal.onnx");
    let res = WatermarkDetector::new(cfg);
    match res {
        Err(AfpError::ModelNotFound(p)) => {
            assert_eq!(p, "/nonexistent/path/to/audioseal.onnx");
        }
        Ok(_) => panic!("expected ModelNotFound, got Ok"),
        Err(e) => panic!("expected ModelNotFound, got Err({e:?})"),
    }
}

#[test]
fn message_bits_above_32_is_config_error() {
    let mut cfg = WatermarkConfig::new("/tmp/dummy.onnx");
    cfg.message_bits = 33;
    match WatermarkDetector::new(cfg) {
        Err(AfpError::Config(msg)) => assert!(msg.contains("message_bits")),
        Ok(_) => panic!("expected Config error, got Ok"),
        Err(e) => panic!("expected Config error, got Err({e:?})"),
    }
}

#[test]
fn threshold_outside_unit_interval_is_config_error() {
    for bad in [-0.1_f32, 1.01, 2.0, -100.0] {
        let mut cfg = WatermarkConfig::new("/tmp/dummy.onnx");
        cfg.threshold = bad;
        match WatermarkDetector::new(cfg) {
            Err(AfpError::Config(msg)) => assert!(msg.contains("threshold")),
            Ok(_) => panic!("expected Config for threshold={bad}, got Ok"),
            Err(e) => panic!("expected Config for threshold={bad}, got Err({e:?})"),
        }
    }
}

#[test]
fn zero_sample_rate_is_config_error() {
    let mut cfg = WatermarkConfig::new("/tmp/dummy.onnx");
    cfg.sample_rate = 0;
    match WatermarkDetector::new(cfg) {
        Err(AfpError::Config(msg)) => assert!(msg.contains("sample_rate")),
        Ok(_) => panic!("expected Config error, got Ok"),
        Err(e) => panic!("expected Config error, got Err({e:?})"),
    }
}

#[test]
fn corrupt_file_returns_model_load_error() {
    use std::io::Write;

    let path = std::env::temp_dir().join(format!(
        "audiofp-watermark-integ-{}.bin",
        std::process::id()
    ));
    {
        let mut f = std::fs::File::create(&path).unwrap();
        f.write_all(&[
            0xDE, 0xAD, 0xBE, 0xEF, 0x00, 0x01, 0x02, 0x03, 0xDE, 0xAD, 0xBE, 0xEF, 0x00, 0x01,
            0x02, 0x03,
        ])
        .unwrap();
    }
    let res = WatermarkDetector::new(WatermarkConfig::new(path.to_string_lossy().into_owned()));
    std::fs::remove_file(&path).ok();
    match res {
        Err(AfpError::ModelLoad(_)) => {}
        Ok(_) => panic!("expected ModelLoad, got Ok"),
        Err(e) => panic!("expected ModelLoad, got Err({e:?})"),
    }
}