summaryrefslogtreecommitdiff
path: root/extractor/mocks/mock_rule_factory.h
blob: 6b7b65864320ddd559935d1ca3328c01c70739a5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#include <gmock/gmock.h>

#include "grammar.h"
#include "rule_factory.h"

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));
};

} // namespace extractor