summaryrefslogtreecommitdiff
path: root/extractor/mocks
diff options
context:
space:
mode:
authorPaul Baltescu <pauldb89@gmail.com>2013-11-26 16:08:41 +0000
committerPaul Baltescu <pauldb89@gmail.com>2013-11-26 16:17:45 +0000
commit7b7167248cd605c9f5bd91b3c87e2826d2336a9f (patch)
tree608c4cf0435225b44a9d34d8f6a7d75cd841ac5a /extractor/mocks
parent3ece9d5cf8c58edd55da35ac96f1a5869d7d6608 (diff)
parent0d1f0af81ad2a368bc3216451a87111be8b9f6f5 (diff)
Merge branch 'master' of github.com:pauldb89/cdec
Diffstat (limited to 'extractor/mocks')
-rw-r--r--extractor/mocks/mock_data_array.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/extractor/mocks/mock_data_array.h b/extractor/mocks/mock_data_array.h
index edc525fa..98e711d2 100644
--- a/extractor/mocks/mock_data_array.h
+++ b/extractor/mocks/mock_data_array.h
@@ -9,6 +9,8 @@ class MockDataArray : public DataArray {
MOCK_CONST_METHOD0(GetData, vector<int>());
MOCK_CONST_METHOD1(AtIndex, int(int index));
MOCK_CONST_METHOD1(GetWordAtIndex, string(int index));
+ MOCK_CONST_METHOD2(GetWordIds, vector<int>(int start_index, int size));
+ MOCK_CONST_METHOD2(GetWords, vector<string>(int start_index, int size));
MOCK_CONST_METHOD0(GetSize, int());
MOCK_CONST_METHOD0(GetVocabularySize, int());
MOCK_CONST_METHOD1(GetWordId, int(const string& word));