jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update images_test.go

Browse Source
This commit is contained in:
Bruce MacDonald 2024-12-03 15:03:08 -08:00
parent 1be080403d
commit a5bc4b7c17
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
server/images_test.go
View File

@ -2,10 +2,10 @@ package server
import (
"context"
"strings"
"testing"
"github.com/ollama/ollama/api"
"github.com/stretchr/testify/assert"
)
func TestPushModel(t *testing.T) {
@ -37,8 +37,14 @@ func TestPushModel(t *testing.T) {
t.Run(tt.modelStr, func(t *testing.T) {
err := PushModel(context.Background(), tt.modelStr, tt.regOpts, noOpProgress)
assert.Error(t, err)
assert.EqualError(t, err, tt.wantErr)
if tt.wantErr != "" {
if err == nil {
t.Errorf("PushModel() error = %v, wantErr %v", err, tt.wantErr)
} else if !strings.Contains(err.Error(), tt.wantErr) {
t.Errorf("PushModel() error = %v, wantErr %v", err, tt.wantErr)
}
return
}
})
}
}