From 5937d8c2b0b039d2c7ab7e85d4bb8655e6173871 Mon Sep 17 00:00:00 2001 From: Kitaiti Makoto Date: Thu, 28 May 2026 22:29:57 +0900 Subject: [PATCH] Add test for Parakeet::Model --- bindings/ruby/test/test_parakeet_model.rb | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 bindings/ruby/test/test_parakeet_model.rb diff --git a/bindings/ruby/test/test_parakeet_model.rb b/bindings/ruby/test/test_parakeet_model.rb new file mode 100644 index 000000000..17cd8ac3c --- /dev/null +++ b/bindings/ruby/test/test_parakeet_model.rb @@ -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