99 lines
2.0 KiB
Go
99 lines
2.0 KiB
Go
|
package python3
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestUnicodeNew(t *testing.T) {
|
||
|
|
Py_Initialize()
|
||
|
|
|
||
|
|
s := PyUnicode_New(20, 'z')
|
||
|
|
assert.NotNil(t, s)
|
||
|
|
defer s.DecRef()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUnicodeFromString(t *testing.T) {
|
||
|
|
Py_Initialize()
|
||
|
|
|
||
|
|
u := PyUnicode_FromString("aaa")
|
||
|
|
assert.True(t, PyUnicode_Check(u))
|
||
|
|
assert.True(t, PyUnicode_CheckExact(u))
|
||
|
|
defer u.DecRef()
|
||
|
|
|
||
|
|
assert.Equal(t, 3, PyUnicode_GetLength(u))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUnicodeFromEncodedObject(t *testing.T) {
|
||
|
|
Py_Initialize()
|
||
|
|
|
||
|
|
b := PyBytes_FromString("bbb")
|
||
|
|
assert.NotNil(t, b)
|
||
|
|
defer b.DecRef()
|
||
|
|
ub := PyUnicode_FromEncodedObject(b, "utf-8", "strict")
|
||
|
|
assert.NotNil(t, ub)
|
||
|
|
defer ub.DecRef()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUnicodeChar(t *testing.T) {
|
||
|
|
Py_Initialize()
|
||
|
|
|
||
|
|
u := PyUnicode_FromString("aaa")
|
||
|
|
assert.True(t, PyUnicode_Check(u))
|
||
|
|
assert.True(t, PyUnicode_CheckExact(u))
|
||
|
|
defer u.DecRef()
|
||
|
|
|
||
|
|
assert.Equal(t, 0, PyUnicode_WriteChar(u, 1, 'd'))
|
||
|
|
|
||
|
|
assert.Equal(t, 'd', PyUnicode_ReadChar(u, 1))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUnicodeFill(t *testing.T) {
|
||
|
|
Py_Initialize()
|
||
|
|
|
||
|
|
u := PyUnicode_FromString("aaa")
|
||
|
|
assert.True(t, PyUnicode_Check(u))
|
||
|
|
assert.True(t, PyUnicode_CheckExact(u))
|
||
|
|
defer u.DecRef()
|
||
|
|
|
||
|
|
assert.Equal(t, 3, PyUnicode_Fill(u, 0, 3, 'c'))
|
||
|
|
|
||
|
|
assert.Equal(t, "ccc", PyUnicode_AsUTF8(u))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUnicodeCopyCharacters(t *testing.T) {
|
||
|
|
Py_Initialize()
|
||
|
|
|
||
|
|
u := PyUnicode_FromString("aaa")
|
||
|
|
assert.True(t, PyUnicode_Check(u))
|
||
|
|
assert.True(t, PyUnicode_CheckExact(u))
|
||
|
|
defer u.DecRef()
|
||
|
|
|
||
|
|
b := PyBytes_FromString("bbb")
|
||
|
|
assert.NotNil(t, b)
|
||
|
|
defer b.DecRef()
|
||
|
|
ub := PyUnicode_FromEncodedObject(b, "utf-8", "strict")
|
||
|
|
assert.NotNil(t, ub)
|
||
|
|
defer ub.DecRef()
|
||
|
|
|
||
|
|
assert.Equal(t, 3, PyUnicode_CopyCharacters(ub, u, 0, 0, 3))
|
||
|
|
|
||
|
|
assert.Equal(t, "aaa", PyUnicode_AsUTF8(ub))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUnicodeSubstring(t *testing.T) {
|
||
|
|
Py_Initialize()
|
||
|
|
|
||
|
|
u := PyUnicode_FromString("aaa")
|
||
|
|
assert.True(t, PyUnicode_Check(u))
|
||
|
|
assert.True(t, PyUnicode_CheckExact(u))
|
||
|
|
defer u.DecRef()
|
||
|
|
|
||
|
|
sub := PyUnicode_Substring(u, 0, 2)
|
||
|
|
assert.NotNil(t, sub)
|
||
|
|
sub.DecRef()
|
||
|
|
|
||
|
|
assert.Equal(t, "aa", PyUnicode_AsUTF8(sub))
|
||
|
|
}
|