diff options
| author | Chris Dyer <redpony@gmail.com> | 2013-11-13 11:22:24 -0800 |
|---|---|---|
| committer | Chris Dyer <redpony@gmail.com> | 2013-11-13 11:22:24 -0800 |
| commit | 1b39e848903743990ca16e2323235b31db20178c (patch) | |
| tree | 19fceda02d09c9d4990aba692ed97193020d1e86 /extractor/grammar_extractor_test.cc | |
| parent | f83186887c94b2ff8b17aefcd0b395f116c09eb6 (diff) | |
| parent | 4c7d24c9357f500839f04c7c8a8cfa0472801e18 (diff) | |
Merge pull request #27 from pks/master
Tidying (soft) syntax features; loo for C++ extractor; updates for dtrain
Diffstat (limited to 'extractor/grammar_extractor_test.cc')
| -rw-r--r-- | extractor/grammar_extractor_test.cc | 7 |
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 |