29 lines
827 B
Ruby
29 lines
827 B
Ruby
require_relative "helper"
|
|
require "stringio"
|
|
|
|
class TestParakeet < TestBase
|
|
def test_log_set
|
|
log_callback = Parakeet.instance_variable_get("@log_callback")
|
|
user_data = Parakeet.instance_variable_get("@log_callback_user_data")
|
|
|
|
$stdout = StringIO.new
|
|
Parakeet.log_set proc {|level, message, _| puts [level, message].join(": ")}, nil
|
|
Parakeet::Context.new("test/fixtures/for-tests-ggml-parakeet-tdt.bin")
|
|
sleep 0.1
|
|
$stdout.rewind
|
|
logs = $stdout.string
|
|
assert_match /loading model from/, logs
|
|
ensure
|
|
$stdout = STDOUT
|
|
Parakeet.log_set log_callback, user_data
|
|
end
|
|
|
|
def test_system_info_str
|
|
assert_match /\APARAKEET : MTL : EMBED_LIBRARY = \d \| CPU : NEON = \d \|/, Parakeet.system_info_str
|
|
end
|
|
|
|
def test_version
|
|
assert_instance_of String, Parakeet::VERSION
|
|
end
|
|
end
|