36 lines
732 B
Go
36 lines
732 B
Go
|
package python3
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestExceptionNew(t *testing.T) {
|
||
|
|
Py_Initialize()
|
||
|
|
|
||
|
|
exc := PyErr_NewException("test_module.TestException", nil, nil)
|
||
|
|
assert.NotNil(t, exc)
|
||
|
|
defer exc.DecRef()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestExceptionNewDoc(t *testing.T) {
|
||
|
|
Py_Initialize()
|
||
|
|
|
||
|
|
exc := PyErr_NewExceptionWithDoc("test_module.TestException", "docstring", nil, nil)
|
||
|
|
assert.NotNil(t, exc)
|
||
|
|
defer exc.DecRef()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestExceptionContext(t *testing.T) {
|
||
|
|
Py_Initialize()
|
||
|
|
|
||
|
|
exc := PyErr_NewException("test_module.TestException", nil, nil)
|
||
|
|
assert.NotNil(t, exc)
|
||
|
|
defer exc.DecRef()
|
||
|
|
|
||
|
|
PyException_SetContext(exc, PyExc_BrokenPipeError)
|
||
|
|
|
||
|
|
assert.Equal(t, PyExc_BrokenPipeError, PyException_GetContext(exc))
|
||
|
|
}
|