Add Version

mirecl · Aug 19, 2024 · eda70b3 · eda70b3
1 parent 9f58838
commit eda70b3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/catboost/catboost.go b/catboost/catboost.go
@@ -50,6 +50,10 @@ var (
 
 var catboostSharedLibraryPath = ""
 
+func Version() string {
+	return fmt.Sprintf("v%d.%d.%d", C.CATBOOST_APPLIER_MAJOR, C.CATBOOST_APPLIER_MINOR, C.CATBOOST_APPLIER_FIX)
+}
+
 func SetSharedLibraryPath(path string) {
 	catboostSharedLibraryPath = path
 }

diff --git a/catboost/catboost_test.go b/catboost/catboost_test.go
@@ -15,6 +15,10 @@ const (
 	testModelPathMetadata            = "../example/metadata/metadata.cbm"
 )
 
+func TestVersion(t *testing.T) {
+	require.Equal(t, "v1.2.5", cb.Version())
+}
+
 func TestLoadFullModel(t *testing.T) {
 	modelRegressor, err := cb.LoadFullModelFromFile(testModelPathRegressor)
 	require.NoError(t, err)