starred/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2026-04-22 16:55:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

x/model: add roundtrip for String test

Browse Source
This commit is contained in:
Blake Mizerany
2024-04-07 16:08:46 -07:00
parent 2100129e83
commit bdff89bc4c
3 changed files with 55 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
x/model/name_test.go
View File

@@ -23,7 +23,7 @@ func fieldsFromName(p Name) fields {
model: p.parts[PartModel],
tag: p.parts[PartTag],
build: p.parts[PartBuild],
digest: p.parts[PartDigest],
digest: p.digest.String(),
}
}
@@ -101,8 +101,8 @@ var testNames = map[string]fields{
func TestNameParts(t *testing.T) {
var p Name
if len(p.Parts()) != int(NumParts) {
t.Errorf("Parts() = %d; want %d", len(p.Parts()), NumParts)
if w, g := int(PartBuild+1), len(p.Parts()); w != g {
t.Errorf("Parts() = %d; want %d", g, w)
}
}
@@ -137,7 +137,7 @@ func TestParseName(t *testing.T) {
// test round-trip
if !ParseName(name.String()).EqualFold(name) {
t.Errorf("String() = %s; want %s", name.String(), baseName)
t.Errorf("ParseName(%q).String() = %s; want %s", s, name.String(), baseName)
}
if name.Valid() && name.DisplayModel() == "" {
@@ -146,9 +146,9 @@ func TestParseName(t *testing.T) {
t.Errorf("Valid() = false; Model() = %q; want empty name", got.model)
}
if name.Resolved() && name.Digest() == "" {
if name.Resolved() && !name.Digest().Valid() {
t.Errorf("Resolved() = true; Digest() = %q; want non-empty digest", got.digest)
} else if !name.Resolved() && name.Digest() != "" {
} else if !name.Resolved() && name.Digest().Valid() {
t.Errorf("Resolved() = false; Digest() = %q; want empty digest", got.digest)
}
})
@@ -233,7 +233,7 @@ func TestNameDisplay(t *testing.T) {
wantLong: "library/mistral:latest",
wantComplete: "example.com/library/mistral:latest",
wantModel: "mistral",
wantGoString: "example.com/library/mistral:latest+Q4_0@?",
wantGoString: "example.com/library/mistral:latest+Q4_0@?-?",
},
{
name: "Short Name",
@@ -242,7 +242,7 @@ func TestNameDisplay(t *testing.T) {
wantLong: "mistral:latest",
wantComplete: "mistral:latest",
wantModel: "mistral",
wantGoString: "?/?/mistral:latest+?@?",
wantGoString: "?/?/mistral:latest+?@?-?",
},
{
name: "Long Name",
@@ -251,7 +251,7 @@ func TestNameDisplay(t *testing.T) {
wantLong: "library/mistral:latest",
wantComplete: "library/mistral:latest",
wantModel: "mistral",
wantGoString: "?/library/mistral:latest+?@?",
wantGoString: "?/library/mistral:latest+?@?-?",
},
{
name: "Case Preserved",
@@ -260,7 +260,7 @@ func TestNameDisplay(t *testing.T) {
wantLong: "Library/Mistral:Latest",
wantComplete: "Library/Mistral:Latest",
wantModel: "Mistral",
wantGoString: "?/Library/Mistral:Latest+?@?",
wantGoString: "?/Library/Mistral:Latest+?@?-?",
},
{
name: "With digest",