初始化
This commit is contained in:
@@ -0,0 +1,24 @@
|
||||
// 提示词模块单元测试,验证固定类型规则和用户输入标准化行为。
|
||||
package prompt
|
||||
|
||||
import "testing"
|
||||
|
||||
// TestIsFixedDramaParseType 验证固定提示词类型及空白清理规则。
|
||||
func TestIsFixedDramaParseType(t *testing.T) {
|
||||
for _, value := range []string{"剧本解析", " 角色、场景、道具解析 "} {
|
||||
if !IsFixedDramaParseType(value) {
|
||||
t.Fatalf("expected %q to be fixed", value)
|
||||
}
|
||||
}
|
||||
if IsFixedDramaParseType("剧本分析") {
|
||||
t.Fatal("剧本分析不应被识别为固定解析提示词")
|
||||
}
|
||||
}
|
||||
|
||||
// TestNormalizeCustomPromptInput 验证提示词字段在校验和写入前统一清理两端空白。
|
||||
func TestNormalizeCustomPromptInput(t *testing.T) {
|
||||
input := normalizeCustomPromptInput(CustomPromptInput{Name: " 名称 ", Type: " 类型\n", Content: " 内容 "})
|
||||
if input.Name != "名称" || input.Type != "类型" || input.Content != "内容" {
|
||||
t.Fatalf("unexpected normalized input: %#v", input)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user