@@ -119,8 +119,12 @@ func TestFindModelFile(t *testing.T) {
119119 dir := t .TempDir ()
120120
121121 // Create a GGUF file
122- os .WriteFile (filepath .Join (dir , "model.gguf" ), []byte ("gguf" ), 0o644 )
123- os .WriteFile (filepath .Join (dir , "config.json" ), []byte ("{}" ), 0o644 )
122+ if err := os .WriteFile (filepath .Join (dir , "model.gguf" ), []byte ("gguf" ), 0o644 ); err != nil {
123+ t .Fatal (err )
124+ }
125+ if err := os .WriteFile (filepath .Join (dir , "config.json" ), []byte ("{}" ), 0o644 ); err != nil {
126+ t .Fatal (err )
127+ }
124128
125129 path , format , err := FindModelFile (dir )
126130 if err != nil {
@@ -137,7 +141,9 @@ func TestFindModelFile(t *testing.T) {
137141func TestFindModelFile_SafeTensors (t * testing.T ) {
138142 dir := t .TempDir ()
139143
140- os .WriteFile (filepath .Join (dir , "model.safetensors" ), []byte ("st" ), 0o644 )
144+ if err := os .WriteFile (filepath .Join (dir , "model.safetensors" ), []byte ("st" ), 0o644 ); err != nil {
145+ t .Fatal (err )
146+ }
141147
142148 path , format , err := FindModelFile (dir )
143149 if err != nil {
@@ -153,7 +159,9 @@ func TestFindModelFile_SafeTensors(t *testing.T) {
153159
154160func TestFindModelFile_NotFound (t * testing.T ) {
155161 dir := t .TempDir ()
156- os .WriteFile (filepath .Join (dir , "config.json" ), []byte ("{}" ), 0o644 )
162+ if err := os .WriteFile (filepath .Join (dir , "config.json" ), []byte ("{}" ), 0o644 ); err != nil {
163+ t .Fatal (err )
164+ }
157165
158166 _ , _ , err := FindModelFile (dir )
159167 if err == nil {
@@ -163,8 +171,12 @@ func TestFindModelFile_NotFound(t *testing.T) {
163171
164172func TestFindModelFile_PicksHighestPrecisionGGUF (t * testing.T ) {
165173 dir := t .TempDir ()
166- os .WriteFile (filepath .Join (dir , "low-Q4_0.gguf" ), []byte ("a" ), 0o644 )
167- os .WriteFile (filepath .Join (dir , "high-Q8_0.gguf" ), []byte ("b" ), 0o644 )
174+ if err := os .WriteFile (filepath .Join (dir , "low-Q4_0.gguf" ), []byte ("a" ), 0o644 ); err != nil {
175+ t .Fatal (err )
176+ }
177+ if err := os .WriteFile (filepath .Join (dir , "high-Q8_0.gguf" ), []byte ("b" ), 0o644 ); err != nil {
178+ t .Fatal (err )
179+ }
168180
169181 path , format , err := FindModelFile (dir )
170182 if err != nil {
0 commit comments