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