29 lines
761 B
Ruby
29 lines
761 B
Ruby
require_relative "helper"
|
|
|
|
class TestVAD < TestBase
|
|
def setup
|
|
@whisper = Whisper::Context.new("base.en")
|
|
vad_params = Whisper::VAD::Params.new
|
|
@params = Whisper::Params.new(
|
|
vad: true,
|
|
vad_model_path: "silero-v6.2.0",
|
|
vad_params:
|
|
)
|
|
end
|
|
|
|
def test_transcribe
|
|
@whisper.transcribe(TestBase::AUDIO, @params) do |text|
|
|
assert_match(/ask not what your country can do for you[,.] ask what you can do for your country/i, text)
|
|
end
|
|
end
|
|
|
|
def test_vad_segments_api
|
|
@whisper.transcribe TestBase::AUDIO, @params
|
|
|
|
assert_equal 4, @whisper.full_n_vad_segments
|
|
@whisper.full_n_vad_segments.times do |i|
|
|
assert @whisper.full_get_vad_segment_t0(i) < @whisper.full_get_vad_segment_t1(i)
|
|
end
|
|
end
|
|
end
|