diff options
| author | Paul Baltescu <pauldb89@gmail.com> | 2013-11-25 23:56:31 +0000 |
|---|---|---|
| committer | Paul Baltescu <pauldb89@gmail.com> | 2013-11-25 23:56:31 +0000 |
| commit | 3c73e472444ff0cd436b12f3679440a6969cbf2d (patch) | |
| tree | 9ceee03648ea671d7f05215826dc0d0a5890e36b /extractor/mocks | |
| parent | 2b95390f08d9f556e6207ecff03b4b0fd5ede993 (diff) | |
Clean up leave-one-out sampling.
Diffstat (limited to 'extractor/mocks')
| -rw-r--r-- | extractor/mocks/mock_rule_factory.h | 6 | ||||
| -rw-r--r-- | extractor/mocks/mock_sampler.h | 4 |
2 files changed, 6 insertions, 4 deletions
diff --git a/extractor/mocks/mock_rule_factory.h b/extractor/mocks/mock_rule_factory.h index 6b7b6586..53eb5022 100644 --- a/extractor/mocks/mock_rule_factory.h +++ b/extractor/mocks/mock_rule_factory.h @@ -7,9 +7,9 @@ namespace extractor { class MockHieroCachingRuleFactory : public HieroCachingRuleFactory { public: - MOCK_METHOD3(GetGrammar, Grammar(const vector<int>& word_ids, const - unordered_set<int>& blacklisted_sentence_ids, - const shared_ptr<DataArray> source_data_array)); + MOCK_METHOD2(GetGrammar, Grammar( + const vector<int>& word_ids, + const unordered_set<int>& blacklisted_sentence_ids)); }; } // namespace extractor diff --git a/extractor/mocks/mock_sampler.h b/extractor/mocks/mock_sampler.h index 75c43c27..b2742f62 100644 --- a/extractor/mocks/mock_sampler.h +++ b/extractor/mocks/mock_sampler.h @@ -7,7 +7,9 @@ namespace extractor { class MockSampler : public Sampler { public: - MOCK_CONST_METHOD1(Sample, PhraseLocation(const PhraseLocation& location)); + MOCK_CONST_METHOD2(Sample, PhraseLocation( + const PhraseLocation& location, + const unordered_set<int>& blacklisted_sentence_ids)); }; } // namespace extractor |