2013-04-10 09:30:11 +00:00
|
|
|
/*
|
|
|
|
* Copyright (C) 2013 The Android Open Source Project
|
|
|
|
*
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
*
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
*
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
* limitations under the License.
|
|
|
|
*/
|
|
|
|
|
|
|
|
package com.android.inputmethod.latin;
|
|
|
|
|
|
|
|
import android.test.AndroidTestCase;
|
|
|
|
import android.test.suitebuilder.annotation.SmallTest;
|
|
|
|
|
2014-05-23 11:18:17 +00:00
|
|
|
import com.android.inputmethod.latin.SuggestedWords.SuggestedWordInfo;
|
2013-06-23 16:11:32 +00:00
|
|
|
|
2013-04-10 09:30:11 +00:00
|
|
|
import java.util.ArrayList;
|
|
|
|
import java.util.Locale;
|
|
|
|
|
|
|
|
@SmallTest
|
|
|
|
public class SuggestedWordsTests extends AndroidTestCase {
|
2014-08-24 21:37:24 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Helper method to create a dummy {@link SuggestedWordInfo} with specifying
|
|
|
|
* {@link SuggestedWordInfo#KIND_TYPED}.
|
|
|
|
*
|
|
|
|
* @param word the word to be used to create {@link SuggestedWordInfo}.
|
|
|
|
* @return a new instance of {@link SuggestedWordInfo}.
|
|
|
|
*/
|
|
|
|
private static SuggestedWordInfo createTypedWordInfo(final String word) {
|
|
|
|
// Use 100 as the frequency because the numerical value does not matter as
|
|
|
|
// long as it's > 1 and < INT_MAX.
|
|
|
|
return new SuggestedWordInfo(word, 100 /* score */,
|
|
|
|
SuggestedWordInfo.KIND_TYPED,
|
|
|
|
null /* sourceDict */,
|
|
|
|
SuggestedWordInfo.NOT_AN_INDEX /* indexOfTouchPointOfSecondWord */,
|
|
|
|
1 /* autoCommitFirstWordConfidence */);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Helper method to create a dummy {@link SuggestedWordInfo} with specifying
|
|
|
|
* {@link SuggestedWordInfo#KIND_CORRECTION}.
|
|
|
|
*
|
|
|
|
* @param word the word to be used to create {@link SuggestedWordInfo}.
|
|
|
|
* @return a new instance of {@link SuggestedWordInfo}.
|
|
|
|
*/
|
|
|
|
private static SuggestedWordInfo createCorrectionWordInfo(final String word) {
|
|
|
|
return new SuggestedWordInfo(word, 1 /* score */,
|
|
|
|
SuggestedWordInfo.KIND_CORRECTION,
|
|
|
|
null /* sourceDict */,
|
|
|
|
SuggestedWordInfo.NOT_AN_INDEX /* indexOfTouchPointOfSecondWord */,
|
|
|
|
SuggestedWordInfo.NOT_A_CONFIDENCE /* autoCommitFirstWordConfidence */);
|
|
|
|
}
|
|
|
|
|
2015-02-25 23:57:59 +00:00
|
|
|
private static ArrayList<SuggestedWordInfo> createCorrectionWordInfos(final String... words) {
|
|
|
|
final ArrayList<SuggestedWordInfo> infos = new ArrayList<>();
|
|
|
|
for (final String word : words) {
|
|
|
|
infos.add(createCorrectionWordInfo(word));
|
|
|
|
}
|
|
|
|
return infos;
|
|
|
|
}
|
|
|
|
|
2013-09-17 10:43:22 +00:00
|
|
|
// Helper for testGetTransformedWordInfo
|
2014-10-20 05:48:56 +00:00
|
|
|
private static SuggestedWordInfo transformWordInfo(final String info,
|
2013-09-17 10:43:22 +00:00
|
|
|
final int trailingSingleQuotesCount) {
|
2014-08-24 21:37:24 +00:00
|
|
|
final SuggestedWordInfo suggestedWordInfo = createTypedWordInfo(info);
|
2013-10-21 05:40:32 +00:00
|
|
|
final SuggestedWordInfo returnedWordInfo =
|
|
|
|
Suggest.getTransformedSuggestedWordInfo(suggestedWordInfo,
|
2013-09-17 10:43:22 +00:00
|
|
|
Locale.ENGLISH, false /* isAllUpperCase */, false /* isFirstCharCapitalized */,
|
|
|
|
trailingSingleQuotesCount);
|
2013-10-21 05:40:32 +00:00
|
|
|
assertEquals(suggestedWordInfo.mAutoCommitFirstWordConfidence,
|
|
|
|
returnedWordInfo.mAutoCommitFirstWordConfidence);
|
|
|
|
return returnedWordInfo;
|
2015-02-25 23:57:59 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public void testRemoveDupesNoDupes() {
|
|
|
|
final ArrayList<SuggestedWordInfo> infos = createCorrectionWordInfos("a", "c");
|
|
|
|
assertEquals(-1, SuggestedWordInfo.removeDups("b", infos));
|
|
|
|
assertEquals(2, infos.size());
|
|
|
|
}
|
|
|
|
|
|
|
|
public void testRemoveDupesTypedWordNotDupe() {
|
|
|
|
final ArrayList<SuggestedWordInfo> infos = createCorrectionWordInfos("a", "a", "c");
|
|
|
|
assertEquals(-1, SuggestedWordInfo.removeDups("b", infos));
|
|
|
|
assertEquals(2, infos.size());
|
|
|
|
}
|
|
|
|
|
|
|
|
public void testRemoveDupesTypedWordOnlyDupe() {
|
|
|
|
final ArrayList<SuggestedWordInfo> infos = createCorrectionWordInfos("a", "b", "c");
|
|
|
|
assertEquals(1, SuggestedWordInfo.removeDups("b", infos));
|
|
|
|
assertEquals(2, infos.size());
|
|
|
|
}
|
|
|
|
|
|
|
|
public void testRemoveDupesTypedWordNotOnlyDupe() {
|
|
|
|
final ArrayList<SuggestedWordInfo> infos = createCorrectionWordInfos("a", "b", "b", "c");
|
|
|
|
assertEquals(1, SuggestedWordInfo.removeDups("b", infos));
|
|
|
|
assertEquals(2, infos.size());
|
2013-09-17 10:43:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public void testGetTransformedSuggestedWordInfo() {
|
|
|
|
SuggestedWordInfo result = transformWordInfo("word", 0);
|
|
|
|
assertEquals(result.mWord, "word");
|
|
|
|
result = transformWordInfo("word", 1);
|
|
|
|
assertEquals(result.mWord, "word'");
|
|
|
|
result = transformWordInfo("word", 3);
|
|
|
|
assertEquals(result.mWord, "word'''");
|
|
|
|
result = transformWordInfo("didn't", 0);
|
|
|
|
assertEquals(result.mWord, "didn't");
|
|
|
|
result = transformWordInfo("didn't", 1);
|
|
|
|
assertEquals(result.mWord, "didn't");
|
|
|
|
result = transformWordInfo("didn't", 3);
|
|
|
|
assertEquals(result.mWord, "didn't''");
|
|
|
|
}
|
2014-08-24 21:37:24 +00:00
|
|
|
|
|
|
|
public void testGetTypedWordInfoOrNull() {
|
|
|
|
final String TYPED_WORD = "typed";
|
2014-11-17 09:26:15 +00:00
|
|
|
final SuggestedWordInfo TYPED_WORD_INFO = createTypedWordInfo(TYPED_WORD);
|
2014-08-24 21:37:24 +00:00
|
|
|
final int NUMBER_OF_ADDED_SUGGESTIONS = 5;
|
|
|
|
final ArrayList<SuggestedWordInfo> list = new ArrayList<>();
|
2014-11-17 09:26:15 +00:00
|
|
|
list.add(TYPED_WORD_INFO);
|
2014-08-24 21:37:24 +00:00
|
|
|
for (int i = 0; i < NUMBER_OF_ADDED_SUGGESTIONS; ++i) {
|
|
|
|
list.add(createCorrectionWordInfo(Integer.toString(i)));
|
|
|
|
}
|
|
|
|
|
|
|
|
// Make sure getTypedWordInfoOrNull() returns non-null object.
|
|
|
|
final SuggestedWords wordsWithTypedWord = new SuggestedWords(
|
|
|
|
list, null /* rawSuggestions */,
|
2014-11-17 09:26:15 +00:00
|
|
|
TYPED_WORD_INFO,
|
2014-08-24 21:37:24 +00:00
|
|
|
false /* typedWordValid */,
|
|
|
|
false /* willAutoCorrect */,
|
|
|
|
false /* isObsoleteSuggestions */,
|
2014-11-07 08:10:36 +00:00
|
|
|
SuggestedWords.INPUT_STYLE_NONE,
|
|
|
|
SuggestedWords.NOT_A_SEQUENCE_NUMBER);
|
2014-08-24 21:37:24 +00:00
|
|
|
final SuggestedWordInfo typedWord = wordsWithTypedWord.getTypedWordInfoOrNull();
|
|
|
|
assertNotNull(typedWord);
|
|
|
|
assertEquals(TYPED_WORD, typedWord.mWord);
|
|
|
|
|
2014-10-08 05:40:03 +00:00
|
|
|
// Make sure getTypedWordInfoOrNull() returns null when no typed word.
|
|
|
|
list.remove(0);
|
|
|
|
final SuggestedWords wordsWithoutTypedWord = new SuggestedWords(
|
|
|
|
list, null /* rawSuggestions */,
|
2014-11-07 08:10:36 +00:00
|
|
|
null /* typedWord */,
|
2014-10-08 05:40:03 +00:00
|
|
|
false /* typedWordValid */,
|
|
|
|
false /* willAutoCorrect */,
|
|
|
|
false /* isObsoleteSuggestions */,
|
2014-11-07 08:10:36 +00:00
|
|
|
SuggestedWords.INPUT_STYLE_NONE,
|
|
|
|
SuggestedWords.NOT_A_SEQUENCE_NUMBER);
|
2014-08-24 21:37:24 +00:00
|
|
|
assertNull(wordsWithoutTypedWord.getTypedWordInfoOrNull());
|
|
|
|
|
|
|
|
// Make sure getTypedWordInfoOrNull() returns null.
|
2014-09-22 03:40:41 +00:00
|
|
|
assertNull(SuggestedWords.getEmptyInstance().getTypedWordInfoOrNull());
|
2014-08-29 04:18:23 +00:00
|
|
|
|
|
|
|
final SuggestedWords emptySuggestedWords = new SuggestedWords(
|
|
|
|
new ArrayList<SuggestedWordInfo>(), null /* rawSuggestions */,
|
2014-11-07 08:10:36 +00:00
|
|
|
null /* typedWord */,
|
2014-08-29 04:18:23 +00:00
|
|
|
false /* typedWordValid */,
|
|
|
|
false /* willAutoCorrect */,
|
|
|
|
false /* isObsoleteSuggestions */,
|
2014-11-07 08:10:36 +00:00
|
|
|
SuggestedWords.INPUT_STYLE_NONE,
|
|
|
|
SuggestedWords.NOT_A_SEQUENCE_NUMBER);
|
2014-08-29 04:18:23 +00:00
|
|
|
assertNull(emptySuggestedWords.getTypedWordInfoOrNull());
|
|
|
|
|
2014-09-22 03:40:41 +00:00
|
|
|
assertNull(SuggestedWords.getEmptyInstance().getTypedWordInfoOrNull());
|
2014-08-24 21:37:24 +00:00
|
|
|
}
|
2013-04-10 09:30:11 +00:00
|
|
|
}
|