am bd0ac979: Fix: native unit test.
* commit 'bd0ac97917284183ea7a59a084cd57b724a854c8': Fix: native unit test.main
commit
aadd46c2db
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue