mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-06-11 04:06:49 +02:00
0251445005
* Prevent overflow
* Fix memsize of Whisper::Context
* Rename xxx_initialize to more Ruby-esque name: xxx_s_new
* Define Whisper::Model::ZipURI
* Define Whisper::Model.coreml_compiled_models
* Make Options' @cmake_options Hash
* Use --{enable,disable}-whisper-coreml option for -I/opt/homebrew/opt/llvm/include
* Prepare Core ML model if enabled
* Add test for ZipURI
* Add signatures for ZipURI
* Add Whisper.system_info_str
* Add test for Whisper.system_info_str
* Add signagure for Model.coreml_compiled_models
* Add signature for Whisper.system_info_str
* Add test for Core ML
* Update date
* Maintain .gitignore
51 lines
1.8 KiB
Ruby
51 lines
1.8 KiB
Ruby
require_relative "helper"
|
|
require 'tempfile'
|
|
require 'tmpdir'
|
|
require 'shellwords'
|
|
|
|
class TestPackage < TestBase
|
|
def test_build
|
|
Tempfile.create do |file|
|
|
assert system("gem", "build", "whispercpp.gemspec", "--output", file.to_path.shellescape, exception: true)
|
|
assert file.size > 0
|
|
assert_path_exist file.to_path
|
|
end
|
|
end
|
|
|
|
sub_test_case "Building binary on installation" do
|
|
def setup
|
|
system "rake", "build", exception: true
|
|
end
|
|
|
|
def test_install
|
|
gemspec = Gem::Specification.load("whispercpp.gemspec")
|
|
Dir.mktmpdir do |dir|
|
|
system "gem", "install", "--install-dir", dir.shellescape, "--no-document", "pkg/#{gemspec.file_name.shellescape}", exception: true
|
|
assert_installed dir, gemspec.version
|
|
end
|
|
end
|
|
|
|
def test_install_with_coreml
|
|
omit_unless RUBY_PLATFORM.match?(/darwin/) do
|
|
gemspec = Gem::Specification.load("whispercpp.gemspec")
|
|
Dir.mktmpdir do |dir|
|
|
system "gem", "install", "--install-dir", dir.shellescape, "--no-document", "pkg/#{gemspec.file_name.shellescape}", "--", "--enable-whisper-coreml", exception: true
|
|
assert_installed dir, gemspec.version
|
|
assert_nothing_raised do
|
|
libdir = File.join(dir, "gems", "#{gemspec.name}-#{gemspec.version}", "lib")
|
|
system "ruby", "-I", libdir, "-r", "whisper", "-e", "Whisper::Context.new('tiny')", exception: true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def assert_installed(dir, version)
|
|
assert_path_exist File.join(dir, "gems/whispercpp-#{version}/lib", "whisper.#{RbConfig::CONFIG["DLEXT"]}")
|
|
assert_path_exist File.join(dir, "gems/whispercpp-#{version}/LICENSE")
|
|
assert_path_not_exist File.join(dir, "gems/whispercpp-#{version}/ext/build")
|
|
end
|
|
end
|
|
end
|