summaryrefslogtreecommitdiff
path: root/extractor/mocks/mock_target_phrase_extractor.h
blob: a8d456310207fd56097cd223650f1ba8064f4cfe (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