diff --git a/integration/basic_test.go b/integration/basic_test.go index 1e3e5c58..dfa921d7 100644 --- a/integration/basic_test.go +++ b/integration/basic_test.go @@ -30,7 +30,24 @@ func TestOrcaMiniBlueSky(t *testing.T) { GenerateTestHelper(ctx, t, req, []string{"rayleigh", "scattering"}) } -func TestUnicodeOutput(t *testing.T) { +func TestUnicode(t *testing.T) { + ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute) + defer cancel() + // Set up the test data + req := api.GenerateRequest{ + // DeepSeek has a Unicode tokenizer regex, making it a unicode torture test + Model: "deepseek-coder-v2:16b-lite-instruct-q2_K", + Prompt: "天空为什么是蓝色的?", + Stream: &stream, + Options: map[string]interface{}{ + "temperature": 0, + "seed": 123, + }, + } + GenerateTestHelper(ctx, t, req, []string{"散射", "频率"}) +} + +func TestExtendedUnicodeOutput(t *testing.T) { ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute) defer cancel() // Set up the test data