extractor/mocks/mock_vocabulary.h


1
2
3
4
5
6
7
8
9
10
11
12
13

#include <gmock/gmock.h>

#include "../vocabulary.h"

namespace extractor {

class MockVocabulary : public Vocabulary {
 public:
  MOCK_METHOD1(GetTerminalValue, string(int word_id));
  MOCK_METHOD1(GetTerminalIndex, int(const string& word));
};

} // namespace extractor