makarna/pkg/quant/config_test.go

package quant

import (
	"testing"
)

func TestMatchPattern(t *testing.T) {
	tests := []struct {
		name     string
		pattern  string
		expected bool
	}{
		{"model.embed_tokens.weight", "*embed_tokens*", true},
		{"model.layers.0.self_attn.v_proj.weight", "*v_proj*", true},
		{"model.layers.0.mlp.down_proj.weight", "*down_proj*", true},
		{"model.norm.weight", "*norm*", true},
		{"model.layers.0.self_attn.q_proj.weight", "*embed_tokens*", false},
	}
	
	for _, tt := range tests {
		result := matchPattern(tt.name, tt.pattern)
		if result != tt.expected {
			t.Errorf("matchPattern(%q, %q) = %v, want %v", tt.name, tt.pattern, result, tt.expected)
		}
	}
}