| 418 | } |
| 419 | |
| 420 | function testCase(inputString, expectedTokenIds) { |
| 421 | const actualTokens = tokenizer.encode(inputString, true, true, true); |
| 422 | if (!isEqual(actualTokens, expectedTokenIds)) { |
| 423 | throw `Test failed. LLaMA Tokenizer Encoder returned unexpected result: expected tokenize(${inputString}) === ${expectedTokenIds}, actual was: ${actualTokens}`; |
| 424 | } |
| 425 | if (inputString !== tokenizer.decode(actualTokens)) { |
| 426 | throw `Test failed. LLaMA Tokenizer Decoder returned unexpected result: expected decode(${actualTokens}) === ${inputString}, actual was: ${decode(actualTokens)}`; |
| 427 | } |
| 428 | } |
| 429 | |
| 430 | // Simple test case |
| 431 | testCase("grabbed", [1, 2646, 1327, 287]); |