cleanup tests

This commit is contained in:
Michael Yang 2024-07-05 16:52:01 -07:00
parent 1954ec5917
commit 78140a712c

View file

@ -6,23 +6,8 @@ import (
"time" "time"
"github.com/google/go-cmp/cmp" "github.com/google/go-cmp/cmp"
"github.com/stretchr/testify/require"
) )
func TestSmoke(t *testing.T) {
t.Setenv("OLLAMA_DEBUG", "")
require.False(t, Debug())
t.Setenv("OLLAMA_DEBUG", "false")
require.False(t, Debug())
t.Setenv("OLLAMA_DEBUG", "1")
require.True(t, Debug())
t.Setenv("OLLAMA_FLASH_ATTENTION", "1")
require.True(t, FlashAttention())
}
func TestHost(t *testing.T) { func TestHost(t *testing.T) {
cases := map[string]struct { cases := map[string]struct {
value string value string