22 lines
449 B
Go
22 lines
449 B
Go
package dialect
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"gpt-load/internal/protocol"
|
|
)
|
|
|
|
func TestNewSetIndexesDialectsByProtocol(t *testing.T) {
|
|
openAI := NewOpenAI()
|
|
got := NewSet(openAI)
|
|
if len(got) != 1 || got[protocol.OpenAICompletions] != openAI {
|
|
t.Fatalf("NewSet(OpenAI) = %#v, want OpenAI indexed", got)
|
|
}
|
|
}
|
|
|
|
func TestNewSetAllowsAnEmptyRuntimeSet(t *testing.T) {
|
|
if got := NewSet(); len(got) != 0 {
|
|
t.Fatalf("NewSet() = %#v, want empty", got)
|
|
}
|
|
}
|