44 lines
1.4 KiB
Dart
44 lines
1.4 KiB
Dart
import 'package:flutter_test/flutter_test.dart';
|
|
import 'package:whisper_ggml/whisper_ggml.dart';
|
|
|
|
void main() {
|
|
group('WhisperModel.modelUri', () {
|
|
test('regular models download from ggerganov/whisper.cpp', () {
|
|
expect(
|
|
WhisperModel.base.modelUri.toString(),
|
|
'https://huggingface.co/ggerganov/whisper.cpp/resolve/main/ggml-base.bin',
|
|
);
|
|
});
|
|
|
|
test('tdrz models download from akashmjn/tinydiarize-whisper.cpp', () {
|
|
expect(
|
|
WhisperModel.smallEnTdrz.modelUri.toString(),
|
|
'https://huggingface.co/akashmjn/tinydiarize-whisper.cpp/resolve/main/ggml-small.en-tdrz.bin',
|
|
);
|
|
});
|
|
});
|
|
|
|
group('WhisperTranscribeSegment.speakerTurnNext', () {
|
|
test('parses speaker_turn_next from native response', () {
|
|
final WhisperTranscribeSegment segment =
|
|
WhisperTranscribeSegment.fromJson(const <String, dynamic>{
|
|
'from_ts': 0,
|
|
'to_ts': 1100,
|
|
'text': 'hello',
|
|
'speaker_turn_next': true,
|
|
});
|
|
expect(segment.speakerTurnNext, isTrue);
|
|
});
|
|
|
|
test('defaults to false when key is absent (older native layer)', () {
|
|
final WhisperTranscribeSegment segment =
|
|
WhisperTranscribeSegment.fromJson(const <String, dynamic>{
|
|
'from_ts': 0,
|
|
'to_ts': 1100,
|
|
'text': 'hello',
|
|
});
|
|
expect(segment.speakerTurnNext, isFalse);
|
|
});
|
|
});
|
|
}
|