Commit 78140a71 authored by Michael Yang's avatar Michael Yang
Browse files

cleanup tests

parent 1954ec59
...@@ -6,23 +6,8 @@ import ( ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment