Skip to content

Commit d2d0249

Browse files
committed
Test with patch
1 parent 33b2a6e commit d2d0249

2 files changed

Lines changed: 10 additions & 1 deletion

File tree

extension/llm/runner/test/test_text_llm_runner.cpp

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,11 @@ class MockTokenizer : public ::tokenizers::Tokenizer {
4343
decode,
4444
(uint64_t, uint64_t),
4545
(const));
46+
MOCK_METHOD(
47+
::tokenizers::Result<std::string>,
48+
id_to_piece,
49+
(uint64_t),
50+
(const));
4651
MOCK_METHOD(uint64_t, bos_tok, (), (const));
4752
MOCK_METHOD(uint64_t, eos_tok, (), (const));
4853
MOCK_METHOD(uint64_t, vocab_size, (), (const));
@@ -127,6 +132,10 @@ class RunnerTest : public Test {
127132
return ::tokenizers::Result<std::string>("token");
128133
});
129134

135+
ON_CALL(*tokenizer, id_to_piece).WillByDefault([](uint64_t) {
136+
return ::tokenizers::Result<std::string>("piece");
137+
});
138+
130139
ON_CALL(*tokenizer, bos_tok()).WillByDefault(Return(1));
131140
ON_CALL(*tokenizer, eos_tok()).WillByDefault(Return(2));
132141
ON_CALL(*tokenizer, vocab_size()).WillByDefault(Return(100));

extension/llm/tokenizers

0 commit comments

Comments
 (0)