Function run_waveform_case

tests/unittests/test_rnnoise_utility.cpp:98–121

}

void run_waveform_case(const engine::audio::RnnoiseModel & model, const std::string & checkpoint, int case_index) {
    const auto fixture = engine::assets::open_tensor_source(
        asset_path("framework/audio_utilities/rnnoise/" + checkpoint + "_waveform_case" + std::to_string(case_index) + ".safetensors"));
    const auto input = fixture->require_f32_tensor("waveform_input");
    const auto expected_output = fixture->require_f32_tensor("waveform_output");
    const auto expected_vad = fixture->require_f32_tensor("waveform_vad");
    require(input.shape.rank == 2 && input.shape.dims[0] == 1, "RNNoise waveform input shape mismatch");
    require(expected_output.shape.rank == 2 && expected_output.shape.dims[0] == 1, "RNNoise waveform output shape mismatch");
    require(expected_vad.shape.rank == 2 && expected_vad.shape.dims[0] == 1, "RNNoise waveform vad shape mismatch");
    const auto actual = model.process_mono_48k(input.values);
    require(actual.sample_rate == 48000, "RNNoise sample rate mismatch");
    require_close(
        actual.samples,
        expected_output.values,
        2.0e-3f,
        2.0e-4,
        checkpoint + " waveform case " + std::to_string(case_index));
    require_close(
        actual.vad,
        expected_vad.values,
        2.5e-5f,
        2.0e-6,
        checkpoint + " waveform vad case " + std::to_string(case_index));
}

}  // namespace