| Index: webrtc/modules/audio_processing/aec3/suppression_gain_unittest.cc
|
| diff --git a/webrtc/modules/audio_processing/aec3/suppression_gain_unittest.cc b/webrtc/modules/audio_processing/aec3/suppression_gain_unittest.cc
|
| index f4feb74a776bf95bc18011bdb2d759ca9a0f39a2..8cfb488dd42f86cdd6e95aa460ab0448c68db571 100644
|
| --- a/webrtc/modules/audio_processing/aec3/suppression_gain_unittest.cc
|
| +++ b/webrtc/modules/audio_processing/aec3/suppression_gain_unittest.cc
|
| @@ -33,7 +33,7 @@ TEST(SuppressionGain, NullOutputGains) {
|
| .GetGain(E2, R2, N2, false,
|
| std::vector<std::vector<float>>(
|
| 3, std::vector<float>(kBlockSize, 0.f)),
|
| - 1, &high_bands_gain, nullptr),
|
| + false, t1, &high_bands_gain, nullptr),
|
| "");
|
| }
|
|
|
| @@ -128,7 +128,8 @@ TEST(SuppressionGain, BasicGainComputation) {
|
| R2.fill(0.1f);
|
| N2.fill(100.f);
|
| for (int k = 0; k < 10; ++k) {
|
| - suppression_gain.GetGain(E2, R2, N2, false, x, 1, &high_bands_gain, &g);
|
| + suppression_gain.GetGain(E2, R2, N2, false, x, 1, false, &high_bands_gain,
|
| + &g);
|
| }
|
| std::for_each(g.begin(), g.end(),
|
| [](float a) { EXPECT_NEAR(1.f, a, 0.001); });
|
| @@ -138,7 +139,8 @@ TEST(SuppressionGain, BasicGainComputation) {
|
| R2.fill(0.1f);
|
| N2.fill(0.f);
|
| for (int k = 0; k < 10; ++k) {
|
| - suppression_gain.GetGain(E2, R2, N2, false, x, 1, &high_bands_gain, &g);
|
| + suppression_gain.GetGain(E2, R2, N2, false, x, 1, false, &high_bands_gain,
|
| + &g);
|
| }
|
| std::for_each(g.begin(), g.end(),
|
| [](float a) { EXPECT_NEAR(1.f, a, 0.001); });
|
| @@ -148,10 +150,15 @@ TEST(SuppressionGain, BasicGainComputation) {
|
| R2.fill(100.f);
|
| N2.fill(0.f);
|
| for (int k = 0; k < 10; ++k) {
|
| - suppression_gain.GetGain(E2, R2, N2, false, x, 1, &high_bands_gain, &g);
|
| + suppression_gain.GetGain(E2, R2, N2, false, x, 1, false, &high_bands_gain,
|
| + &g);
|
| }
|
| std::for_each(g.begin(), g.end(),
|
| [](float a) { EXPECT_NEAR(0.f, a, 0.001); });
|
| +
|
| + // Verify the functionality for forcing a zero gain.
|
| + suppression_gain.GetGain(E2, R2, N2, false, x, 1, true, &high_bands_gain, &g);
|
| + std::for_each(g.begin(), g.end(), [](float a) { EXPECT_FLOAT_EQ(0.f, a); });
|
| }
|
|
|
| } // namespace aec3
|
|
|