Fix: NPE in unit test.

Change-Id: I769b43f2afa1151d5c0d9c408018c3932be377dc
This commit is contained in:
Keisuke Kuroyanagi 2014-10-03 19:53:23 +09:00
parent 80861cde9d
commit 2522a4a477

View file

@ -1237,15 +1237,17 @@ public class BinaryDictionaryTests extends AndroidTestCase {
wordSet.remove(word0); wordSet.remove(word0);
final HashSet<String> bigramWord1s = bigrams.get(word0); final HashSet<String> bigramWord1s = bigrams.get(word0);
// TODO: Support ngram. // TODO: Support ngram.
for (final WeightedString bigramTarget : wordProperty.getBigrams()) { if (wordProperty.mHasNgrams) {
final String word1 = bigramTarget.mWord; for (final WeightedString bigramTarget : wordProperty.getBigrams()) {
assertTrue(bigramWord1s.contains(word1)); final String word1 = bigramTarget.mWord;
final Pair<String, String> bigram = new Pair<>(word0, word1); assertTrue(bigramWord1s.contains(word1));
if (canCheckBigramProbability(formatVersion)) { final Pair<String, String> bigram = new Pair<>(word0, word1);
final int bigramProbability = bigramProbabilitiesToCheckLater.get(bigram); if (canCheckBigramProbability(formatVersion)) {
assertEquals(bigramProbability, bigramTarget.getProbability()); final int bigramProbability = bigramProbabilitiesToCheckLater.get(bigram);
assertEquals(bigramProbability, bigramTarget.getProbability());
}
bigramSet.remove(bigram);
} }
bigramSet.remove(bigram);
} }
token = result.mNextToken; token = result.mNextToken;
} while (token != 0); } while (token != 0);