From a5bc4b7c1776fa80c119621b9e9a78fb8bac5d5a Mon Sep 17 00:00:00 2001 From: Bruce MacDonald Date: Tue, 3 Dec 2024 15:03:08 -0800 Subject: [PATCH] Update images_test.go --- server/images_test.go | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/server/images_test.go b/server/images_test.go index 34410cf2e..cdec48a3c 100644 --- a/server/images_test.go +++ b/server/images_test.go @@ -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 + } }) } }