chromium/components/live_caption/greedy_text_stabilizer_unittest.cc

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/live_caption/greedy_text_stabilizer.h"

#include <string>
#include <vector>

#include "testing/gtest/include/gtest/gtest.h"

namespace captions {

class GreedyTextStabilizerTest : public testing::Test { … };

// Ensures that invalid min token frequency gets set to 0.
TEST_F(GreedyTextStabilizerTest, NegativeMinTokenFrequencySetToZero) { … }

// Tests that empty sentences are handled correctly.
TEST_F(GreedyTextStabilizerTest, EmptySentencesHaveZeroStableTokens) { … }

// Tests that complete sentences are handled correctly.
TEST_F(GreedyTextStabilizerTest, CompleteSentencesOutputFullText) { … }

// Use the same phrase and only change the punctuation.
// The punctuation will change if it is seen more often than others (the mode).
TEST_F(GreedyTextStabilizerTest, PunctuationFlickerIsStabilized) { … }

// Tests the behavior for varying values of the min_token_frequency.
TEST_F(GreedyTextStabilizerTest, MultipleFormsOfFlickerAreStabilized) { … }

// Test that greedy text stabilization works for long text sequences.
TEST_F(GreedyTextStabilizerTest, FlickerInLongFormTextIsStabilized) { … }

TEST_F(GreedyTextStabilizerTest, KoreanTokensAreStabilized) { … }

}  // namespace captions