| Index: webrtc/modules/audio_processing/utility/block_mean_calculator_unittest.cc
|
| diff --git a/webrtc/modules/audio_processing/utility/block_mean_calculator_unittest.cc b/webrtc/modules/audio_processing/utility/block_mean_calculator_unittest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..613437e3da559282358ffacd43e3b257c4dc3892
|
| --- /dev/null
|
| +++ b/webrtc/modules/audio_processing/utility/block_mean_calculator_unittest.cc
|
| @@ -0,0 +1,55 @@
|
| +/*
|
| + * Copyright (c) 2016 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 "testing/gtest/include/gtest/gtest.h"
|
| +
|
| +#include "webrtc/modules/audio_processing/utility/block_mean_calculator.h"
|
| +
|
| +namespace webrtc {
|
| +
|
| +TEST(MeanCalculatorTest, Correctness) {
|
| + const size_t kBlockLength = 10;
|
| + BlockMeanCalculator mean_calculator(kBlockLength);
|
| + size_t i = 0;
|
| + float reference = 0.0;
|
| +
|
| + for (; i < kBlockLength; ++i) {
|
| + EXPECT_EQ(mean_calculator.SamplesSinceLastUpdate(), i % kBlockLength);
|
| + mean_calculator.AddSample(static_cast<float>(i));
|
| + }
|
| + for (; i < 3 * kBlockLength; ++i) {
|
| + const size_t new_samples = i % kBlockLength;
|
| + if (new_samples == 0) {
|
| + // Sum of (i - kBlockLength) ... (i - 1)
|
| + reference = i - 0.5 * (1 + kBlockLength);
|
| + }
|
| + EXPECT_EQ(mean_calculator.SamplesSinceLastUpdate(), new_samples);
|
| + EXPECT_EQ(reference, mean_calculator.GetLatestMean());
|
| + mean_calculator.AddSample(static_cast<float>(i));
|
| + }
|
| +}
|
| +
|
| +TEST(MeanCalculatorTest, Reset) {
|
| + const size_t kBlockLength = 10;
|
| + BlockMeanCalculator mean_calculator(kBlockLength);
|
| + for (size_t i = 0; i < kBlockLength - 1; ++i) {
|
| + mean_calculator.AddSample(static_cast<float>(i));
|
| + }
|
| + EXPECT_EQ(mean_calculator.SamplesSinceLastUpdate(), kBlockLength - 1);
|
| + mean_calculator.Reset();
|
| + for (size_t i = 0; i < kBlockLength; ++i) {
|
| + EXPECT_EQ(mean_calculator.SamplesSinceLastUpdate(), i);
|
| + mean_calculator.AddSample(static_cast<float>(i));
|
| + }
|
| + EXPECT_EQ(mean_calculator.SamplesSinceLastUpdate(), 0u);
|
| + EXPECT_EQ(mean_calculator.GetLatestMean(), 0.5 * (kBlockLength - 1));
|
| +}
|
| +
|
| +} // namespace webrtc
|
|
|