Merge "[TestPrep17] Allow fake dictionary for testing"
commit
a04fbce412
|
@ -18,6 +18,7 @@ package com.android.inputmethod.research;
|
||||||
|
|
||||||
import android.util.Log;
|
import android.util.Log;
|
||||||
|
|
||||||
|
import com.android.inputmethod.annotations.UsedForTesting;
|
||||||
import com.android.inputmethod.latin.Dictionary;
|
import com.android.inputmethod.latin.Dictionary;
|
||||||
import com.android.inputmethod.latin.Suggest;
|
import com.android.inputmethod.latin.Suggest;
|
||||||
import com.android.inputmethod.latin.define.ProductionFlag;
|
import com.android.inputmethod.latin.define.ProductionFlag;
|
||||||
|
@ -64,7 +65,11 @@ public abstract class MainLogBuffer extends FixedLogBuffer {
|
||||||
// The size of the n-grams logged. E.g. N_GRAM_SIZE = 2 means to sample bigrams.
|
// The size of the n-grams logged. E.g. N_GRAM_SIZE = 2 means to sample bigrams.
|
||||||
public static final int N_GRAM_SIZE = 2;
|
public static final int N_GRAM_SIZE = 2;
|
||||||
|
|
||||||
|
// TODO: Remove dependence on Suggest, and pass in Dictionary as a parameter to an appropriate
|
||||||
|
// method.
|
||||||
private Suggest mSuggest;
|
private Suggest mSuggest;
|
||||||
|
@UsedForTesting
|
||||||
|
private Dictionary mDictionaryForTesting;
|
||||||
private boolean mIsStopping = false;
|
private boolean mIsStopping = false;
|
||||||
|
|
||||||
/* package for test */ int mNumWordsBetweenNGrams;
|
/* package for test */ int mNumWordsBetweenNGrams;
|
||||||
|
@ -83,7 +88,15 @@ public abstract class MainLogBuffer extends FixedLogBuffer {
|
||||||
mSuggest = suggest;
|
mSuggest = suggest;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@UsedForTesting
|
||||||
|
/* package for test */ void setDictionaryForTesting(final Dictionary dictionary) {
|
||||||
|
mDictionaryForTesting = dictionary;
|
||||||
|
}
|
||||||
|
|
||||||
private Dictionary getDictionary() {
|
private Dictionary getDictionary() {
|
||||||
|
if (mDictionaryForTesting != null) {
|
||||||
|
return mDictionaryForTesting;
|
||||||
|
}
|
||||||
if (mSuggest == null || !mSuggest.hasMainDictionary()) return null;
|
if (mSuggest == null || !mSuggest.hasMainDictionary()) return null;
|
||||||
return mSuggest.getMainDictionary();
|
return mSuggest.getMainDictionary();
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue