config_test.go 470 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
package envconfig

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestConfig(t *testing.T) {
Daniel Hiltgen's avatar
Daniel Hiltgen committed
10
11
	Debug = false // Reset whatever was loaded in init()
	t.Setenv("OLLAMA_DEBUG", "")
12
13
	LoadConfig()
	require.False(t, Debug)
Daniel Hiltgen's avatar
Daniel Hiltgen committed
14
	t.Setenv("OLLAMA_DEBUG", "false")
15
16
	LoadConfig()
	require.False(t, Debug)
Daniel Hiltgen's avatar
Daniel Hiltgen committed
17
	t.Setenv("OLLAMA_DEBUG", "1")
18
19
	LoadConfig()
	require.True(t, Debug)
20
21
22
	t.Setenv("OLLAMA_FLASH_ATTENTION", "1")
	LoadConfig()
	require.True(t, FlashAttention)
23
}