add unicode tests

Author: cpunion

unicode.go

@@ -4,10 +4,7 @@ package gp
 #include <Python.h>
 */
 import "C"
-import (
-	"reflect"
-	"unsafe"
-)
+import "unsafe"
 
 type Str struct {
 	Object
@@ -18,9 +15,8 @@ func newStr(obj *PyObject) Str {
 }
 
 func MakeStr(s string) Str {
-	hdr := (*reflect.StringHeader)(unsafe.Pointer(&s))
-	ptr := (*C.char)(unsafe.Pointer(hdr.Data))
-	length := C.long(hdr.Len)
+	ptr := (*C.char)(unsafe.Pointer(unsafe.StringData(s)))
+	length := C.long(len(s))
 	return newStr(C.PyUnicode_FromStringAndSize(ptr, length))
 }
 
@@ -31,6 +27,10 @@ func (s Str) String() string {
 }
 
 func (s Str) Len() int {
+	return int(C.PyUnicode_GetLength(s.obj))
+}
+
+func (s Str) ByteLen() int {
 	var l C.long
 	C.PyUnicode_AsUTF8AndSize(s.obj, &l)
 	return int(l)

unicode_test.go

@@ -0,0 +1,103 @@
+package gp
+
+import (
+	"testing"
+)
+
+func TestMakeStr(t *testing.T) {
+	tests := []struct {
+		name      string
+		input     string
+		expected  string
+		length    int
+		byteCount int
+	}{
+		{
+			name:      "empty string",
+			input:     "",
+			expected:  "",
+			length:    0,
+			byteCount: 0,
+		},
+		{
+			name:      "ascii string",
+			input:     "hello",
+			expected:  "hello",
+			length:    5,
+			byteCount: 5, // ASCII字符每个占1字节
+		},
+		{
+			name:      "unicode string",
+			input:     "你好世界",
+			expected:  "你好世界",
+			length:    4,
+			byteCount: 12, // 中文字符每个占3字节
+		},
+		{
+			name:      "mixed string",
+			input:     "hello世界",
+			expected:  "hello世界",
+			length:    7,
+			byteCount: 11, // 5个ASCII字符(5字节) + 2个中文字符(6字节)
+		},
+		{
+			name:      "special unicode",
+			input:     "π∑€",
+			expected:  "π∑€",
+			length:    3,
+			byteCount: 8, // π(2字节) + ∑(3字节) + €(3字节) = 8字节
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			pyStr := MakeStr(tt.input)
+
+			// Test String() method
+			if got := pyStr.String(); got != tt.expected {
+				t.Errorf("MakeStr(%q).String() = %q, want %q", tt.input, got, tt.expected)
+			}
+
+			// Test Len() method
+			if got := pyStr.Len(); got != tt.length {
+				t.Errorf("MakeStr(%q).Len() = %d, want %d", tt.input, got, tt.length)
+			}
+
+			// Test ByteLen() method
+			if got := pyStr.ByteLen(); got != tt.byteCount {
+				t.Errorf("MakeStr(%q).ByteLen() = %d, want %d", tt.input, got, tt.byteCount)
+			}
+		})
+	}
+}
+
+func TestStrEncode(t *testing.T) {
+	tests := []struct {
+		name     string
+		input    string
+		encoding string
+	}{
+		{
+			name:     "utf-8 encoding",
+			input:    "hello世界",
+			encoding: "utf-8",
+		},
+		{
+			name:     "ascii encoding",
+			input:    "hello",
+			encoding: "ascii",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			pyStr := MakeStr(tt.input)
+			encoded := pyStr.Encode(tt.encoding)
+			decoded := encoded.Decode(tt.encoding)
+
+			if got := decoded.String(); got != tt.input {
+				t.Errorf("String encode/decode roundtrip failed: got %q, want %q", got, tt.input)
+			}
+		})
+	}
+}