Unit test update for I7ed95a54344d8b.

Change-Id: Ic8af56b8741ee7c231d73ce6a660fd32c9f5a3b3
main
Keisuke Kuroyanagi 2014-03-12 15:41:20 +09:00
parent 5dc17c22ca
commit 61467744e7
1 changed files with 8 additions and 2 deletions

View File

@ -948,10 +948,13 @@ public class BinaryDictionaryTests extends AndroidTestCase {
final HashSet<String> bigramWord1s = bigrams.get(word0);
final WordProperty wordProperty = binaryDictionary.getWordProperty(word0);
assertEquals(bigramWord1s.size(), wordProperty.mBigrams.size());
final int unigramProbability = wordProperty.getProbability();
for (int j = 0; j < wordProperty.mBigrams.size(); j++) {
final String word1 = wordProperty.mBigrams.get(j).mWord;
assertTrue(bigramWord1s.contains(word1));
final int probability = wordProperty.mBigrams.get(j).getProbability();
final int bigramProbability = wordProperty.mBigrams.get(j).getProbability();
final int probability = binaryDictionary.calculateProbability(
unigramProbability, bigramProbability);
assertEquals((int)bigramProbabilities.get(new Pair<String, String>(word0, word1)),
probability);
assertEquals(wordProperty.mBigrams.get(j).getProbability(), probability);
@ -1037,11 +1040,14 @@ public class BinaryDictionaryTests extends AndroidTestCase {
assertEquals((int)wordProbabilitiesToCheckLater.get(word0),
wordProperty.mProbabilityInfo.mProbability);
wordSet.remove(word0);
final int unigramProbability = wordProperty.getProbability();
final HashSet<String> bigramWord1s = bigrams.get(word0);
for (int j = 0; j < wordProperty.mBigrams.size(); j++) {
final String word1 = wordProperty.mBigrams.get(j).mWord;
assertTrue(bigramWord1s.contains(word1));
final int probability = wordProperty.mBigrams.get(j).getProbability();
final int bigramProbability = wordProperty.mBigrams.get(j).getProbability();
final int probability = binaryDictionary.calculateProbability(
unigramProbability, bigramProbability);
final Pair<String, String> bigram = new Pair<String, String>(word0, word1);
assertEquals((int)bigramProbabilitiesToCheckLater.get(bigram), probability);
bigramSet.remove(bigram);