summaryrefslogtreecommitdiff
path: root/extractor/grammar_extractor_test.cc
diff options
context:
space:
mode:
authorPatrick Simianer <p@simianer.de>2013-11-13 18:00:10 +0100
committerPatrick Simianer <p@simianer.de>2013-11-13 18:00:10 +0100
commit2d2d5eced93d58bc77894d8c328195cd9950b96d (patch)
tree41892c0943a0049611ea6035e181cb9cbd1c7d19 /extractor/grammar_extractor_test.cc
parent2d025c839e474045d81b7490adc8842ad427c4e1 (diff)
unit tests for extractor loo sampling
Diffstat (limited to 'extractor/grammar_extractor_test.cc')
-rw-r--r--extractor/grammar_extractor_test.cc7
1 files changed, 5 insertions, 2 deletions
diff --git a/extractor/grammar_extractor_test.cc b/extractor/grammar_extractor_test.cc
index 823bb8b4..f32a9599 100644
--- a/extractor/grammar_extractor_test.cc
+++ b/extractor/grammar_extractor_test.cc
@@ -39,12 +39,15 @@ TEST(GrammarExtractorTest, TestAnnotatingWords) {
vector<Rule> rules;
vector<string> feature_names;
Grammar grammar(rules, feature_names);
- EXPECT_CALL(*factory, GetGrammar(word_ids))
+ unordered_set<int> blacklisted_sentence_ids;
+ shared_ptr<DataArray> source_data_array;
+ EXPECT_CALL(*factory, GetGrammar(word_ids, blacklisted_sentence_ids, source_data_array))
.WillOnce(Return(grammar));
GrammarExtractor extractor(vocabulary, factory);
string sentence = "Anna has many many apples .";
- extractor.GetGrammar(sentence);
+
+ extractor.GetGrammar(sentence, blacklisted_sentence_ids, source_data_array);
}
} // namespace