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

#include "target_phrase_extractor.h"

namespace extractor {

typedef pair<Phrase, PhraseAlignment> PhraseExtract;

class MockTargetPhraseExtractor : public TargetPhraseExtractor {
 public:
  MOCK_CONST_METHOD6(ExtractPhrases, vector<PhraseExtract>(
      const vector<pair<int, int> > &, const vector<int>&, int, int,
      const unordered_map<int, int>&, int));
};

} // namespace extractor