Updated some tests

Author: djthorpe

pkg/whisper/whisper_test.go

@@ -19,7 +19,7 @@ import (
 const MODEL_TINY = "ggml-tiny.en-q5_1.bin"
 const SAMPLE_EN = "../../samples/jfk.wav"
 const SAMPLE_FR = "../../samples/OlivierL.wav"
-const SAMPLE_DE = "../../samples/ge-podcast.wav"
+const SAMPLE_DE = "../../samples/de-podcast.wav"
 
 func Test_whisper_001(t *testing.T) {
 	assert := assert.New(t)
@@ -225,7 +225,7 @@ func Test_whisper_005(t *testing.T) {
 
 		assert.NoError(service.WithModel(model, func(task *task.Context) error {
 			t.Log("Transcribing", len(samples), "samples")
-			return task.Transcribe(context.Background(), 0, samples, false, nil)
+			return task.Transcribe(context.Background(), 0, samples, nil)
 		}))
 	})
 
@@ -241,7 +241,7 @@ func Test_whisper_005(t *testing.T) {
 
 		assert.NoError(service.WithModel(model, func(task *task.Context) error {
 			t.Log("Transcribing", len(samples), "samples")
-			return task.Transcribe(context.Background(), 0, samples, false, nil)
+			return task.Transcribe(context.Background(), 0, samples, nil)
 		}))
 	})
 
@@ -257,7 +257,7 @@ func Test_whisper_005(t *testing.T) {
 
 		assert.NoError(service.WithModel(model, func(task *task.Context) error {
 			t.Log("Transcribing", len(samples), "samples")
-			return task.Transcribe(context.Background(), 0, samples, false, nil)
+			return task.Transcribe(context.Background(), 0, samples, nil)
 		}))
 	})
 }
@@ -294,7 +294,7 @@ func Test_whisper_006(t *testing.T) {
 
 			assert.NoError(service.WithModel(model, func(task *task.Context) error {
 				t.Log("Transcribing", len(samples), "samples")
-				return task.Transcribe(context.Background(), 0, samples, false, nil)
+				return task.Transcribe(context.Background(), 0, samples, nil)
 			}))
 		})
 
@@ -312,7 +312,7 @@ func Test_whisper_006(t *testing.T) {
 
 			assert.NoError(service.WithModel(model, func(task *task.Context) error {
 				t.Log("Transcribing", len(samples), "samples")
-				return task.Transcribe(context.Background(), 0, samples, false, nil)
+				return task.Transcribe(context.Background(), 0, samples, nil)
 			}))
 		})
 
@@ -330,7 +330,7 @@ func Test_whisper_006(t *testing.T) {
 
 			assert.NoError(service.WithModel(model, func(task *task.Context) error {
 				t.Log("Transcribing", len(samples), "samples")
-				return task.Transcribe(context.Background(), 0, samples, false, nil)
+				return task.Transcribe(context.Background(), 0, samples, nil)
 			}))
 		})
 	})

sys/whisper/whisper_test.go

@@ -15,7 +15,7 @@ import (
 
 const SAMPLE_EN = "../../samples/jfk.wav"
 const SAMPLE_FR = "../../samples/OlivierL.wav"
-const SAMPLE_DE = "../../samples/ge-podcast.wav"
+const SAMPLE_DE = "../../samples/de-podcast.wav"
 
 func Test_whisper_00(t *testing.T) {
 	assert := assert.New(t)
@@ -482,7 +482,7 @@ func Test_whisper_05(t *testing.T) {
 		params.SetLanguage("auto")
 		params.SetTranslate(false)
 		params.SetTokenTimestamps(true)
-		params.SetDiarizeEnable(true)
+		params.SetDiarize(true)
 		params.SetSegmentCallback(ctx, func(new_segments int) {
 			num_segments := ctx.NumSegments()
 			for i := num_segments - new_segments; i < num_segments; i++ {
@@ -518,7 +518,7 @@ func Test_whisper_05(t *testing.T) {
 		params.SetLanguage("auto")
 		params.SetTranslate(false)
 		params.SetTokenTimestamps(true)
-		params.SetDiarizeEnable(true)
+		params.SetDiarize(true)
 		params.SetSegmentCallback(ctx, func(new_segments int) {
 			num_segments := ctx.NumSegments()
 			for i := num_segments - new_segments; i < num_segments; i++ {