Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit bd0ac979 authored by Keisuke Kuroyanagi's avatar Keisuke Kuroyanagi
Browse files

Fix: native unit test. 

Bug: 14425059
Change-Id: I7c5103eac35e01a5732f1f4594d730df06dfca96
parent c912957f

native/jni/tests/suggest/policyimpl/dictionary/structure/v4/content/language_model_dict_content_test.cpp

+4 −4
Original line number Diff line number Diff line
@@ -107,15 +107,15 @@ TEST(LanguageModelDictContentTest, TestGetWordProbability) { 
    languageModelDictContent.setProbabilityEntry(prevWordIds[0], &probabilityEntry);

    languageModelDictContent.setNgramProbabilityEntry(prevWordIds.limit(1), wordId,

            &bigramProbabilityEntry);

    EXPECT_EQ(bigramProbability, languageModelDictContent.getWordProbability(prevWordIds, wordId,

            nullptr /* headerPolicy */));

    EXPECT_EQ(bigramProbability, languageModelDictContent.getWordAttributes(prevWordIds, wordId,

            nullptr /* headerPolicy */).getProbability());

    const ProbabilityEntry trigramProbabilityEntry(flag, trigramProbability);

    languageModelDictContent.setNgramProbabilityEntry(prevWordIds.limit(1),

            prevWordIds[1], &probabilityEntry);

    languageModelDictContent.setNgramProbabilityEntry(prevWordIds.limit(2), wordId,

            &trigramProbabilityEntry);

    EXPECT_EQ(trigramProbability, languageModelDictContent.getWordProbability(prevWordIds, wordId,

            nullptr /* headerPolicy */));

    EXPECT_EQ(trigramProbability, languageModelDictContent.getWordAttributes(prevWordIds, wordId,

            nullptr /* headerPolicy */).getProbability());

}



}  // namespace