21 lines
628 B
Ruby
21 lines
628 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(File.join(__dir__, "../../../models/parakeet-tdt-0.6b-v3.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
|
|
end
|