ExternalVendorCode/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2024-12-24 06:46:37 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Group tests for #transcribe

Browse Source
This commit is contained in:
Kitaiti Makoto 2024-10-18 01:29:50 +09:00
parent 81e6df3bab
commit 71b65b00cc
1 changed files with 29 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
bindings/ruby/tests/test_whisper.rb
View File

@ -116,38 +116,38 @@ class TestWhisper < Test::Unit::TestCase
assert !@params.split_on_word
end
def test_whisper
@whisper = Whisper::Context.new(File.join(TOPDIR, '..', '..', 'models', 'ggml-base.en.bin'))
params = Whisper::Params.new
params.print_timestamps = false
sub_test_case "#transcribe" do
def setup
@whisper = Whisper::Context.new(File.join(TOPDIR, '..', '..', 'models', 'ggml-base.en.bin'))
@params = Whisper::Params.new
@params.print_timestamps = false
@jfk = File.join(TOPDIR, '..', '..', 'samples', 'jfk.wav')
end
jfk = File.join(TOPDIR, '..', '..', 'samples', 'jfk.wav')
@whisper.transcribe(jfk, params) {|text|
assert_match /ask not what your country can do for you, ask what you can do for your country/, text
}
end
def test_whisper
@whisper.transcribe(@jfk, @params) {|text|
assert_match /ask not what your country can do for you, ask what you can do for your country/, text
}
end
def test_new_segment_callback_lambda
counter = 0
@params.new_segment_callback = ->(text, start_time, end_time, index) {
assert_kind_of String, text
assert_kind_of Integer, start_time
assert_kind_of Integer, end_time
assert_same index, counter
counter += 1
}
whisper = Whisper::Context.new(File.join(TOPDIR, '..', '..', 'models', 'ggml-base.en.bin'))
jfk = File.join(TOPDIR, '..', '..', 'samples', 'jfk.wav')
whisper.transcribe(jfk, @params)
end
def test_new_segment_callback_lambda
counter = 0
@params.new_segment_callback = ->(text, start_time, end_time, index) {
assert_kind_of String, text
assert_kind_of Integer, start_time
assert_kind_of Integer, end_time
assert_same index, counter
counter += 1
}
@whisper.transcribe(@jfk, @params)
end
def test_new_segment_callback_proc
@params.new_segment_callback = proc {|text| # proc checks arguments loosly
assert_kind_of String, text
}
whisper = Whisper::Context.new(File.join(TOPDIR, '..', '..', 'models', 'ggml-base.en.bin'))
jfk = File.join(TOPDIR, '..', '..', 'samples', 'jfk.wav')
whisper.transcribe(jfk, @params)
def test_new_segment_callback_proc
@params.new_segment_callback = proc {|text| # proc checks arguments loosly
assert_kind_of String, text
}
@whisper.transcribe(@jfk, @params)
end
end
def test_build