extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2024-11-07 08:34:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add tests for Params#new_segment_callback=

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

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

@ -127,6 +127,29 @@ class TestWhisper < Test::Unit::TestCase
}
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_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)
end
def test_build
Tempfile.create do |file|
assert system("gem", "build", "whispercpp.gemspec", "--output", file.to_path.shellescape, exception: true)