oww-rs 0.0.1 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

mod tests {
    use ndarray::{arr2, Ix2};
    use crate::audio::AudioFeatures;
    use crate::CHUNK;

    #[test]
    fn test_mels() {
        let mut audio = AudioFeatures::new();

        let sample_data = [0f32; 64000];
        let mels = audio.get_melspectrogram(&arr2(&[sample_data])).unwrap();
        let dim = mels.clone().into_dimensionality::<Ix2>().unwrap().dim();
        assert_eq!(dim, (397, 32));
    }

    #[test]
    fn test_embeddings() {
        let mut audio = AudioFeatures::new();
        let sample_data = [0f32; CHUNK];
        let embeddings = audio.get_audio_features(sample_data.to_vec()).unwrap();
        let dim = embeddings.into_dimensionality::<Ix2>().unwrap().dim();
        assert_eq!(dim, (41, 96))
    }
}