Skip to content

Commit bf1753a

Browse files
committed
Add unit-tests
1 parent 75e6709 commit bf1753a

File tree

1 file changed

+28
-0
lines changed

1 file changed

+28
-0
lines changed

catboost/catboost_test.go

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,9 @@
11
package catboost_test
22

33
import (
4+
"errors"
45
"fmt"
6+
"runtime"
57
"testing"
68

79
cb "github.com/mirecl/catboost-cgo/catboost"
@@ -164,3 +166,29 @@ func TestGetError(t *testing.T) {
164166
err = cb.GetError()
165167
require.NoError(t, err)
166168
}
169+
170+
func TestGetSupportedEvaluatorTypes(t *testing.T) {
171+
// init test model
172+
model, err := cb.LoadFullModelFromFile(testModelPathRegressor)
173+
require.NoError(t, err)
174+
175+
devices, err := model.GetSupportedEvaluatorTypes()
176+
require.NoError(t, err)
177+
require.True(t, len(devices) > 0)
178+
}
179+
180+
func TestEnableGPUEvaluation(t *testing.T) {
181+
// init test model
182+
model, err := cb.LoadFullModelFromFile(testModelPathRegressor)
183+
require.NoError(t, err)
184+
185+
err = model.EnableGPUEvaluation()
186+
187+
if runtime.GOOS == "darwin" {
188+
require.True(t, errors.Is(err, cb.ErrNotSupportedGPU))
189+
}
190+
191+
if runtime.GOOS == "linux" {
192+
require.NoError(t, err)
193+
}
194+
}

0 commit comments

Comments
 (0)