97 lines
2.6 KiB
Go
97 lines
2.6 KiB
Go
package filler
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/eric/seriatim/internal/model"
|
|
)
|
|
|
|
func TestApplyTagsVerySafeFillers(t *testing.T) {
|
|
for _, text := range []string{"um", "uhhh", "ER", "ermm", "ah", "eh", "hmmm", "mm", "mmm"} {
|
|
t.Run(text, func(t *testing.T) {
|
|
got, tagged := Apply(transcript(segment(text, 1, 1.5)))
|
|
if tagged != 1 {
|
|
t.Fatalf("tagged = %d, want 1", tagged)
|
|
}
|
|
assertCategories(t, got.Segments[0], []string{Category})
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestApplyTagsRepeatedFillers(t *testing.T) {
|
|
got, tagged := Apply(transcript(segment("um uh hmm", 1, 1.8)))
|
|
if tagged != 1 {
|
|
t.Fatalf("tagged = %d, want 1", tagged)
|
|
}
|
|
assertCategories(t, got.Segments[0], []string{Category})
|
|
}
|
|
|
|
func TestApplyMatchesTrimAwareCaseInsensitive(t *testing.T) {
|
|
got, tagged := Apply(transcript(segment(" UM uh ", 1, 1.5)))
|
|
if tagged != 1 {
|
|
t.Fatalf("tagged = %d, want 1", tagged)
|
|
}
|
|
assertCategories(t, got.Segments[0], []string{Category})
|
|
}
|
|
|
|
func TestApplyDoesNotTagNonMatches(t *testing.T) {
|
|
for _, text := range []string{"um okay", "uh-huh", "hmm, okay"} {
|
|
t.Run(text, func(t *testing.T) {
|
|
got, tagged := Apply(transcript(segment(text, 1, 1.5)))
|
|
if tagged != 0 {
|
|
t.Fatalf("tagged = %d, want 0", tagged)
|
|
}
|
|
assertCategories(t, got.Segments[0], nil)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestApplyRejectsWordCountOverThree(t *testing.T) {
|
|
got, tagged := Apply(transcript(segment("um uh er ah", 1, 1.5)))
|
|
if tagged != 0 {
|
|
t.Fatalf("tagged = %d, want 0", tagged)
|
|
}
|
|
assertCategories(t, got.Segments[0], nil)
|
|
}
|
|
|
|
func TestApplyRejectsDurationOverOneSecond(t *testing.T) {
|
|
got, tagged := Apply(transcript(segment("um", 1, 2.1)))
|
|
if tagged != 0 {
|
|
t.Fatalf("tagged = %d, want 0", tagged)
|
|
}
|
|
assertCategories(t, got.Segments[0], nil)
|
|
}
|
|
|
|
func TestApplyPreservesExistingCategoriesAndAvoidsDuplicate(t *testing.T) {
|
|
existing := segment("um", 1, 1.2)
|
|
existing.Categories = []string{"manual", Category}
|
|
|
|
got, tagged := Apply(transcript(existing))
|
|
if tagged != 0 {
|
|
t.Fatalf("tagged = %d, want 0", tagged)
|
|
}
|
|
assertCategories(t, got.Segments[0], []string{"manual", Category})
|
|
}
|
|
|
|
func transcript(segments ...model.Segment) model.MergedTranscript {
|
|
return model.MergedTranscript{Segments: segments}
|
|
}
|
|
|
|
func segment(text string, start float64, end float64) model.Segment {
|
|
return model.Segment{
|
|
Source: "input.json",
|
|
Speaker: "Alice",
|
|
Start: start,
|
|
End: end,
|
|
Text: text,
|
|
}
|
|
}
|
|
|
|
func assertCategories(t *testing.T, segment model.Segment, want []string) {
|
|
t.Helper()
|
|
if !reflect.DeepEqual(segment.Categories, want) {
|
|
t.Fatalf("categories = %v, want %v", segment.Categories, want)
|
|
}
|
|
}
|