54 lines
2.0 KiB
Go
54 lines
2.0 KiB
Go
|
package memoir
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
type recoveringEditor struct{ calls int }
|
||
|
|
|
||
|
|
func (g *recoveringEditor) GenerateJSON(ctx context.Context, input, prompt string) (json.RawMessage, error) {
|
||
|
|
g.calls++
|
||
|
|
if g.calls == 1 {
|
||
|
|
return json.RawMessage(`{"broken":`), nil
|
||
|
|
}
|
||
|
|
return (testEditor{}).GenerateJSON(ctx, input, prompt)
|
||
|
|
}
|
||
|
|
func TestMemoirRetriesInvalidProviderJSON(t *testing.T) {
|
||
|
|
g := &recoveringEditor{}
|
||
|
|
s := Service{Generator: g}
|
||
|
|
sources := []Source{{ID: "a", Title: "进厂", Content: "1970年我进厂。", EventTime: "1970年"}}
|
||
|
|
d, e := s.generateBook(context.Background(), sources)
|
||
|
|
if e != nil || g.calls != 2 || len(d.Chapters) != 1 {
|
||
|
|
t.Fatalf("calls=%d doc=%v err=%v", g.calls, d, e)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParseTranscriptUsesSpeakerLabels(t *testing.T) {
|
||
|
|
turns, e := ParseTranscript("我：1970年我进厂。\n还有一行。\n豆包：你当时是不是开飞船？\n我：今天做了晚饭。")
|
||
|
|
if e != nil || len(turns) != 3 || turns[0].Speaker != "subject" || !strings.Contains(turns[0].Content, "还有一行") || turns[1].Speaker != "agent" {
|
||
|
|
t.Fatalf("turns=%v err=%v", turns, e)
|
||
|
|
}
|
||
|
|
if _, e = ParseTranscript("没有说话人标签的记录"); e == nil {
|
||
|
|
t.Fatal("unattributed speech accepted")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func TestDocumentRequiresEverySourceExactlyOnce(t *testing.T) {
|
||
|
|
sources := []Source{{ID: "a", EventTime: "1970年"}, {ID: "b", EventTime: "1973年"}}
|
||
|
|
d := Document{Title: "回忆录", Chapters: []Chapter{{StoryID: "a", EventTime: "1970年", Title: "进厂", Paragraphs: []string{"我进厂了。"}, IllustrationPrompt: "车间插画"}, {StoryID: "b", EventTime: "1973年", Title: "搬家", Paragraphs: []string{"我搬家了。"}, IllustrationPrompt: "搬家插画"}}}
|
||
|
|
if e := ValidateDocument(d, sources); e != nil {
|
||
|
|
t.Fatal(e)
|
||
|
|
}
|
||
|
|
d.Chapters[1].StoryID = "a"
|
||
|
|
if ValidateDocument(d, sources) == nil {
|
||
|
|
t.Fatal("duplicate source accepted")
|
||
|
|
}
|
||
|
|
d.Chapters[1].StoryID = "b"
|
||
|
|
d.Chapters[1].EventTime = "2026年"
|
||
|
|
if ValidateDocument(d, sources) == nil {
|
||
|
|
t.Fatal("invented event date accepted")
|
||
|
|
}
|
||
|
|
}
|