mirror of
https://github.com/likelovewant/ollama-for-amd.git
synced 2025-12-21 22:33:56 +00:00
348b3e0983
This commit copies (without history) the bmizerany/ollama-go repository with the intention of integrating it into the ollama as a replacement for the pushing, and pulling of models, and management of the cache they are pushed and pulled from. New homes for these packages will be determined as they are integrated and we have a better understanding of proper package boundaries.
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package testutil
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// Check calls t.Fatal(err) if err is not nil.
|
|
func Check(t *testing.T, err error) {
|
|
if err != nil {
|
|
t.Helper()
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
// CheckFunc exists so other packages do not need to invent their own type for
|
|
// taking a Check function.
|
|
type CheckFunc func(err error)
|
|
|
|
// Checker returns a check function that
|
|
// calls t.Fatal if err is not nil.
|
|
func Checker(t *testing.T) (check func(err error)) {
|
|
return func(err error) {
|
|
if err != nil {
|
|
t.Helper()
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// StopPanic runs f but silently recovers from any panic f causes.
|
|
// The normal usage is:
|
|
//
|
|
// testutil.StopPanic(func() {
|
|
// callThatShouldPanic()
|
|
// t.Errorf("callThatShouldPanic did not panic")
|
|
// })
|
|
func StopPanic(f func()) {
|
|
defer func() { recover() }()
|
|
f()
|
|
}
|
|
|
|
// CheckTime calls t.Fatalf if got != want. Included in the error message is
|
|
// want.Sub(got) to help diagnose the difference, along with their values in
|
|
// UTC.
|
|
func CheckTime(t *testing.T, got, want time.Time) {
|
|
t.Helper()
|
|
if !got.Equal(want) {
|
|
t.Fatalf("got %v, want %v (%v)", got.UTC(), want.UTC(), want.Sub(got))
|
|
}
|
|
}
|
|
|
|
// WriteFile writes data to a file named name. It makes the directory if it
|
|
// doesn't exist and sets the file mode to perm.
|
|
//
|
|
// The name must be a relative path and must not contain .. or start with a /;
|
|
// otherwise WriteFile will panic.
|
|
func WriteFile[S []byte | string](t testing.TB, name string, data S) {
|
|
t.Helper()
|
|
|
|
if filepath.IsAbs(name) {
|
|
t.Fatalf("WriteFile: name must be a relative path, got %q", name)
|
|
}
|
|
name = filepath.Clean(name)
|
|
dir := filepath.Dir(name)
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(name, []byte(data), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|