mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 06:10:40 +01:00
bb1a28de3c
- test_utils.h/.cc simplified - webrtc::rnnvad::test -> webrtc::rnnvad - all unit test code inside the anonymous namespace - names improved Bug: webrtc:10480 Change-Id: I0a0f056f9728bb8a1b93006b95d7ed5bf5bd4adb Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/196509 Commit-Queue: Alessio Bazzica <alessiob@webrtc.org> Reviewed-by: Sam Zackrisson <saza@webrtc.org> Cr-Commit-Position: refs/heads/master@{#32789}
53 lines
2.1 KiB
C++
53 lines
2.1 KiB
C++
/*
|
|
* Copyright (c) 2018 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "modules/audio_processing/agc2/rnn_vad/pitch_search.h"
|
|
|
|
#include <algorithm>
|
|
#include <vector>
|
|
|
|
#include "modules/audio_processing/agc2/cpu_features.h"
|
|
#include "modules/audio_processing/agc2/rnn_vad/pitch_search_internal.h"
|
|
#include "modules/audio_processing/agc2/rnn_vad/test_utils.h"
|
|
// TODO(bugs.webrtc.org/8948): Add when the issue is fixed.
|
|
// #include "test/fpe_observer.h"
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
namespace rnn_vad {
|
|
|
|
// Checks that the computed pitch period is bit-exact and that the computed
|
|
// pitch gain is within tolerance given test input data.
|
|
TEST(RnnVadTest, PitchSearchWithinTolerance) {
|
|
ChunksFileReader reader = CreateLpResidualAndPitchInfoReader();
|
|
const int num_frames = std::min(reader.num_chunks, 300); // Max 3 s.
|
|
std::vector<float> lp_residual(kBufSize24kHz);
|
|
float expected_pitch_period, expected_pitch_strength;
|
|
const AvailableCpuFeatures cpu_features = GetAvailableCpuFeatures();
|
|
PitchEstimator pitch_estimator(cpu_features);
|
|
{
|
|
// TODO(bugs.webrtc.org/8948): Add when the issue is fixed.
|
|
// FloatingPointExceptionObserver fpe_observer;
|
|
for (int i = 0; i < num_frames; ++i) {
|
|
SCOPED_TRACE(i);
|
|
ASSERT_TRUE(reader.reader->ReadChunk(lp_residual));
|
|
ASSERT_TRUE(reader.reader->ReadValue(expected_pitch_period));
|
|
ASSERT_TRUE(reader.reader->ReadValue(expected_pitch_strength));
|
|
int pitch_period =
|
|
pitch_estimator.Estimate({lp_residual.data(), kBufSize24kHz});
|
|
EXPECT_EQ(expected_pitch_period, pitch_period);
|
|
EXPECT_NEAR(expected_pitch_strength,
|
|
pitch_estimator.GetLastPitchStrengthForTesting(), 15e-6f);
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace rnn_vad
|
|
} // namespace webrtc
|