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

#include "phrase_location.h"
#include "matchings_sampler.h"

namespace extractor {

class MockMatchingsSampler : public MatchingsSampler {
 public:
  MOCK_CONST_METHOD2(Sample, PhraseLocation(
      const PhraseLocation& location,
      const unordered_set<int>& blacklisted_sentence_ids));
};

} // namespace extractor