101 lines
2.7 KiB
Ruby
101 lines
2.7 KiB
Ruby
require_relative "helper"
|
|
|
|
class TestVADContext < TestBase
|
|
def test_initialize
|
|
context = Whisper::VAD::Context.new("silero-v6.2.0")
|
|
assert_instance_of Whisper::VAD::Context, context
|
|
end
|
|
|
|
def test_detect
|
|
context = Whisper::VAD::Context.new("silero-v6.2.0")
|
|
segments = context.detect(AUDIO, Whisper::VAD::Params.new)
|
|
assert_segments segments
|
|
end
|
|
|
|
def test_invalid_model_type
|
|
assert_raise TypeError do
|
|
Whisper::VAD::Context.new(Object.new)
|
|
end
|
|
end
|
|
|
|
def test_allocate
|
|
vad = Whisper::VAD::Context.allocate
|
|
assert_raise do
|
|
vad.detect(AUDIO, Whisper::VAD::Params.new)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def assert_segments(segments)
|
|
assert_instance_of Whisper::VAD::Segments, segments
|
|
|
|
i = 0
|
|
segments.each do |segment|
|
|
i += 1
|
|
assert_instance_of Whisper::VAD::Segment, segment
|
|
end
|
|
assert i > 0
|
|
|
|
segments.each_with_index do |segment, index|
|
|
assert_instance_of Integer, index
|
|
end
|
|
|
|
assert_instance_of Enumerator, segments.each
|
|
|
|
segment = segments.each.first
|
|
assert_instance_of Float, segment.start_time
|
|
assert_instance_of Float, segment.end_time
|
|
|
|
segment => {start_time:, end_time:}
|
|
assert_equal segment.start_time, start_time
|
|
assert_equal segment.end_time, end_time
|
|
|
|
assert_equal 4, segments.length
|
|
end
|
|
|
|
sub_test_case "from samples" do
|
|
def setup
|
|
super
|
|
@vad = Whisper::VAD::Context.new("silero-v6.2.0")
|
|
@samples = File.read(AUDIO, nil, 78).unpack("s<*").collect {|i| i.to_f / 2**15}
|
|
end
|
|
|
|
def test_segments_from_samples
|
|
segments = @vad.segments_from_samples(Whisper::VAD::Params.new, @samples, @samples.length)
|
|
assert_segments segments
|
|
end
|
|
|
|
def test_segments_from_samples_without_length
|
|
segments = @vad.segments_from_samples(Whisper::VAD::Params.new, @samples)
|
|
assert_segments segments
|
|
end
|
|
|
|
def test_segments_from_samples_enumerator
|
|
samples = @samples.each
|
|
segments = @vad.segments_from_samples(Whisper::VAD::Params.new, samples, @samples.length)
|
|
assert_segments segments
|
|
end
|
|
|
|
def test_segments_from_samples_enumerator_without_length
|
|
samples = @samples.each
|
|
assert_raise ArgumentError do
|
|
@vad.segments_from_samples(Whisper::VAD::Params.new, samples)
|
|
end
|
|
end
|
|
|
|
def test_segments_from_samples_enumerator_with_too_large_length
|
|
samples = @samples.each.take(10).to_enum
|
|
assert_raise StopIteration do
|
|
@vad.segments_from_samples(Whisper::VAD::Params.new, samples, 11)
|
|
end
|
|
end
|
|
|
|
def test_segments_from_samples_with_memory_view
|
|
samples = JFKReader.new(AUDIO)
|
|
segments = @vad.segments_from_samples(Whisper::VAD::Params.new, samples)
|
|
assert_segments segments
|
|
end
|
|
end
|
|
end
|