First AI generated version
This commit is contained in:
@@ -0,0 +1,49 @@
|
||||
package gollama_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
gollama "gitea.neitzel.de/konrad/go-ollama"
|
||||
)
|
||||
|
||||
func TestOptionSetMerge(t *testing.T) {
|
||||
base := gollama.NewOptions().Temperature(0.8).TopP(0.9)
|
||||
override := gollama.NewOptions().Temperature(0.2)
|
||||
merged := base.Merge(override)
|
||||
|
||||
m, err := merged.ToMapForTest()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if m["temperature"] != float64(0.2) {
|
||||
t.Fatalf("temperature = %v", m["temperature"])
|
||||
}
|
||||
if m["top_p"] != float64(0.9) {
|
||||
t.Fatalf("top_p = %v", m["top_p"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseModelfileParameters(t *testing.T) {
|
||||
params := "temperature 0.5\nnum_ctx 4096\n"
|
||||
opts, err := gollama.ParseModelfileParametersForTest(params)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
m, err := opts.ToMapForTest()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if m["temperature"] != float64(0.5) {
|
||||
t.Fatalf("temperature = %v", m["temperature"])
|
||||
}
|
||||
if m["num_ctx"] != float64(4096) {
|
||||
t.Fatalf("num_ctx = %v", m["num_ctx"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestKnownOptions(t *testing.T) {
|
||||
docs := gollama.KnownOptions()
|
||||
if len(docs) == 0 {
|
||||
t.Fatal("expected known options")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user