tcsenpai/ollama
1
0
Fork 0
mirror of https://github.com/tcsenpai/ollama.git synced 2025-06-07 11:45:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

harden integration tests

Browse Source
This commit is contained in:
Daniel Hiltgen 2024-08-01 14:41:23 -07:00
parent 46c5f5fd9e
commit 343aba9fca
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
integration/concurrency_test.go
View File

@ -40,7 +40,7 @@ func TestMultiModelConcurrency(t *testing.T) {
} }
resp = [2][]string{ resp = [2][]string{
[]string{"sunlight"}, []string{"sunlight"},
[]string{"england", "english", "massachusetts", "pilgrims", "british"}, []string{"england", "english", "massachusetts", "pilgrims", "british", "festival"},
} }
) )
var wg sync.WaitGroup var wg sync.WaitGroup

2
integration/utils_test.go
View File

@ -275,7 +275,7 @@ func DoGenerate(ctx context.Context, t *testing.T, client *api.Client, genReq ap
break break
} }
} }
require.True(t, atLeastOne, "none of %v found in %s", anyResp, response) require.True(t, atLeastOne, "%s: none of %v found in %s", genReq.Model, anyResp, response)
slog.Info("test pass", "model", genReq.Model, "prompt", genReq.Prompt, "contains", anyResp, "response", response) slog.Info("test pass", "model", genReq.Model, "prompt", genReq.Prompt, "contains", anyResp, "response", response)
case <-ctx.Done(): case <-ctx.Done():
t.Error("outer test context done while waiting for generate") t.Error("outer test context done while waiting for generate")