diff --git a/tests/src/com/android/inputmethod/latin/BinaryDictionaryTests.java b/tests/src/com/android/inputmethod/latin/BinaryDictionaryTests.java index a26c25886..65c70a6db 100644 --- a/tests/src/com/android/inputmethod/latin/BinaryDictionaryTests.java +++ b/tests/src/com/android/inputmethod/latin/BinaryDictionaryTests.java @@ -993,6 +993,10 @@ public class BinaryDictionaryTests extends AndroidTestCase { for (final String word : words) { final UnigramProperty unigramProperty = binaryDictionary.getUnigramProperty(word); assertEquals((int)unigramProbabilities.get(word), unigramProperty.mProbability); + if (!shortcutTargets.containsKey(word)) { + // The word does not have shortcut targets. + continue; + } assertEquals(shortcutTargets.get(word).size(), unigramProperty.mShortcutTargets.size()); for (final WeightedString shortcutTarget : unigramProperty.mShortcutTargets) { final String targetCodePonts = shortcutTarget.mWord;