test: custom enum type to string

northes · Apr 23, 2024 · d897175 · d897175
1 parent f3d15e1
commit d897175
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/enum_chat_completions_test.go b/enum_chat_completions_test.go
@@ -0,0 +1,23 @@
+package moonshot_test
+
+import (
+	"testing"
+
+	"github.com/northes/go-moonshot"
+	"github.com/stretchr/testify/require"
+)
+
+func TestEnumChatCompletions(t *testing.T) {
+	tt := require.New(t)
+
+	tt.EqualValues(moonshot.RoleSystem, moonshot.RoleSystem.String())
+	tt.EqualValues(moonshot.RoleUser, moonshot.RoleUser.String())
+	tt.EqualValues(moonshot.RoleAssistant, moonshot.RoleAssistant.String())
+
+	tt.EqualValues(moonshot.ModelMoonshotV18K, moonshot.ModelMoonshotV18K.String())
+	tt.EqualValues(moonshot.ModelMoonshotV132K, moonshot.ModelMoonshotV132K.String())
+	tt.EqualValues(moonshot.ModelMoonshotV1128K, moonshot.ModelMoonshotV1128K.String())
+
+	tt.EqualValues(moonshot.FinishReasonStop, moonshot.FinishReasonStop.String())
+	tt.EqualValues(moonshot.FinishReasonLength, moonshot.FinishReasonLength.String())
+}
diff --git a/enum_files_test.go b/enum_files_test.go
@@ -0,0 +1,14 @@
+package moonshot_test
+
+import (
+	"testing"
+
+	"github.com/northes/go-moonshot"
+	"github.com/stretchr/testify/require"
+)
+
+func TestEnumFiles(t *testing.T) {
+	tt := require.New(t)
+
+	tt.EqualValues(moonshot.FilePurposeExtract, moonshot.FilePurposeExtract.String())
+}