Add test for Parakeet::Model

This commit is contained in:
Kitaiti Makoto 2026-05-28 22:29:57 +09:00
parent 6cefb43f62
commit 5937d8c2b0
1 changed files with 21 additions and 0 deletions

View File

@ -0,0 +1,21 @@
require_relative "helper"
class TestModel < TestBase
def test_model
parakeet = Parakeet::Context.new("test/fixtures/for-tests-ggml-parakeet-tdt.bin")
assert_instance_of Parakeet::Model, parakeet.model
end
def test_attributes
parakeet = Parakeet::Context.new("test/fixtures/for-tests-ggml-parakeet-tdt.bin")
model = parakeet.model
assert_equal 10, model.n_vocab
assert_equal 3200, model.n_audio_ctx
assert_equal 8, model.n_audio_state
assert_equal 2, model.n_audio_head
assert_equal 1, model.n_audio_layer
assert_equal 16, model.n_mels
assert_equal 0, model.ftype
end
end