Fix: native unit test.
Bug: 14425059 Change-Id: I7c5103eac35e01a5732f1f4594d730df06dfca96main
parent
c912957f3f
commit
bd0ac97917
|
@ -107,15 +107,15 @@ TEST(LanguageModelDictContentTest, TestGetWordProbability) {
|
||||||
languageModelDictContent.setProbabilityEntry(prevWordIds[0], &probabilityEntry);
|
languageModelDictContent.setProbabilityEntry(prevWordIds[0], &probabilityEntry);
|
||||||
languageModelDictContent.setNgramProbabilityEntry(prevWordIds.limit(1), wordId,
|
languageModelDictContent.setNgramProbabilityEntry(prevWordIds.limit(1), wordId,
|
||||||
&bigramProbabilityEntry);
|
&bigramProbabilityEntry);
|
||||||
EXPECT_EQ(bigramProbability, languageModelDictContent.getWordProbability(prevWordIds, wordId,
|
EXPECT_EQ(bigramProbability, languageModelDictContent.getWordAttributes(prevWordIds, wordId,
|
||||||
nullptr /* headerPolicy */));
|
nullptr /* headerPolicy */).getProbability());
|
||||||
const ProbabilityEntry trigramProbabilityEntry(flag, trigramProbability);
|
const ProbabilityEntry trigramProbabilityEntry(flag, trigramProbability);
|
||||||
languageModelDictContent.setNgramProbabilityEntry(prevWordIds.limit(1),
|
languageModelDictContent.setNgramProbabilityEntry(prevWordIds.limit(1),
|
||||||
prevWordIds[1], &probabilityEntry);
|
prevWordIds[1], &probabilityEntry);
|
||||||
languageModelDictContent.setNgramProbabilityEntry(prevWordIds.limit(2), wordId,
|
languageModelDictContent.setNgramProbabilityEntry(prevWordIds.limit(2), wordId,
|
||||||
&trigramProbabilityEntry);
|
&trigramProbabilityEntry);
|
||||||
EXPECT_EQ(trigramProbability, languageModelDictContent.getWordProbability(prevWordIds, wordId,
|
EXPECT_EQ(trigramProbability, languageModelDictContent.getWordAttributes(prevWordIds, wordId,
|
||||||
nullptr /* headerPolicy */));
|
nullptr /* headerPolicy */).getProbability());
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
Loading…
Reference in New Issue