nyancrimew
/
LatinIME
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge "Fix unit tests."

main
Keisuke Kuroyanagi 2014-03-14 05:12:50 +00:00 committed by Android (Google) Code Review
parent 5a1e351751 98705b6bf5
commit 4ba6eec9fd
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/src/com/android/inputmethod/latin/BinaryDictionaryTests.java
View File

@ -948,16 +948,15 @@ public class BinaryDictionaryTests extends AndroidTestCase {
final HashSet<String> bigramWord1s = bigrams.get(word0); final HashSet<String> bigramWord1s = bigrams.get(word0);
final WordProperty wordProperty = binaryDictionary.getWordProperty(word0); final WordProperty wordProperty = binaryDictionary.getWordProperty(word0);
assertEquals(bigramWord1s.size(), wordProperty.mBigrams.size()); assertEquals(bigramWord1s.size(), wordProperty.mBigrams.size());
final int unigramProbability = wordProperty.getProbability();
for (int j = 0; j < wordProperty.mBigrams.size(); j++) { for (int j = 0; j < wordProperty.mBigrams.size(); j++) {
final String word1 = wordProperty.mBigrams.get(j).mWord; final String word1 = wordProperty.mBigrams.get(j).mWord;
assertTrue(bigramWord1s.contains(word1)); assertTrue(bigramWord1s.contains(word1));
final int bigramProbability = wordProperty.mBigrams.get(j).getProbability(); final int bigramProbabilityDelta = bigramProbabilities.get(
final int probability = binaryDictionary.calculateProbability( new Pair<String, String>(word0, word1));
unigramProbability, bigramProbability); final int unigramProbability = wordProbabilities.get(word1);
assertEquals((int)bigramProbabilities.get(new Pair<String, String>(word0, word1)), final int bigramProbablity = binaryDictionary.calculateProbability(
probability); unigramProbability, bigramProbabilityDelta);
assertEquals(wordProperty.mBigrams.get(j).getProbability(), probability); assertEquals(wordProperty.mBigrams.get(j).getProbability(), bigramProbablity);
} }
} }
} }
@ -1040,16 +1039,16 @@ public class BinaryDictionaryTests extends AndroidTestCase {
assertEquals((int)wordProbabilitiesToCheckLater.get(word0), assertEquals((int)wordProbabilitiesToCheckLater.get(word0),
wordProperty.mProbabilityInfo.mProbability); wordProperty.mProbabilityInfo.mProbability);
wordSet.remove(word0); wordSet.remove(word0);
final int unigramProbability = wordProperty.getProbability();
final HashSet<String> bigramWord1s = bigrams.get(word0); final HashSet<String> bigramWord1s = bigrams.get(word0);
for (int j = 0; j < wordProperty.mBigrams.size(); j++) { for (int j = 0; j < wordProperty.mBigrams.size(); j++) {
final String word1 = wordProperty.mBigrams.get(j).mWord; final String word1 = wordProperty.mBigrams.get(j).mWord;
assertTrue(bigramWord1s.contains(word1)); assertTrue(bigramWord1s.contains(word1));
final int bigramProbability = wordProperty.mBigrams.get(j).getProbability(); final int unigramProbability = wordProbabilitiesToCheckLater.get(word1);
final int probability = binaryDictionary.calculateProbability(
unigramProbability, bigramProbability);
final Pair<String, String> bigram = new Pair<String, String>(word0, word1); final Pair<String, String> bigram = new Pair<String, String>(word0, word1);
assertEquals((int)bigramProbabilitiesToCheckLater.get(bigram), probability); final int bigramProbabilityDelta = bigramProbabilitiesToCheckLater.get(bigram);
final int bigramProbablity = binaryDictionary.calculateProbability(
unigramProbability, bigramProbabilityDelta);
assertEquals(wordProperty.mBigrams.get(j).getProbability(), bigramProbablity);
bigramSet.remove(bigram); bigramSet.remove(bigram);
} }
token = result.mNextToken; token = result.mNextToken;