Unit tests (#11)

Add unit test

Author: remicalixte

bytes_test.go

@@ -54,9 +54,11 @@ func TestBytesConcat(t *testing.T) {
 
 	bytes1 := PyBytes_FromString(s1)
 
-	bytes2 := PyBytes_FromString(s2)
+	array := PyByteArray_FromStringAndSize(s2)
+	defer array.DecRef()
+
+	bytes2 := PyBytes_FromObject(array)
 	assert.NotNil(t, bytes2)
-	defer bytes2.DecRef()
 
 	bytes1 = PyBytes_Concat(bytes1, bytes2)
 	assert.NotNil(t, bytes1)

complex_test.go

@@ -14,6 +14,8 @@ import (
 )
 
 func TestComplex(t *testing.T) {
+	Py_Initialize()
+
 	real := 2.
 	imaginary := 5.
 

errors_test.go

@@ -0,0 +1,255 @@
+package python3
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestErrorSetString(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_SetString(PyExc_BaseException, "test message")
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorSetObject(t *testing.T) {
+	Py_Initialize()
+
+	message := PyUnicode_FromString("test message")
+	defer message.DecRef()
+
+	PyErr_SetObject(PyExc_BaseException, message)
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Print()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorSetNone(t *testing.T) {
+	Py_Initialize()
+
+	message := PyUnicode_FromString("test message")
+	defer message.DecRef()
+
+	PyErr_SetNone(PyExc_BaseException)
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Print()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorSetObjectEx(t *testing.T) {
+	Py_Initialize()
+
+	message := PyUnicode_FromString("test message")
+	defer message.DecRef()
+
+	PyErr_SetObject(PyExc_BaseException, message)
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_PrintEx(false)
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorWriteUnraisable(t *testing.T) {
+	Py_Initialize()
+
+	message := PyUnicode_FromString("unraisable exception")
+	defer message.DecRef()
+
+	PyErr_WriteUnraisable(message)
+
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorBadArgument(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_BadArgument()
+
+	assert.NotNil(t, PyErr_Occurred())
+
+	PyErr_Clear()
+
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorNoMemory(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_NoMemory()
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorBadInternalCall(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_BadInternalCall()
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorImportError(t *testing.T) {
+	Py_Initialize()
+
+	message := PyUnicode_FromString("test message")
+	defer message.DecRef()
+
+	PyErr_SetImportError(message, nil, nil)
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorImportErrorSubclass(t *testing.T) {
+	Py_Initialize()
+
+	message := PyUnicode_FromString("test message")
+	defer message.DecRef()
+
+	PyErr_SetImportErrorSubclass(message, nil, nil, Dict)
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorSyntax(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_SetNone(PyExc_SyntaxError)
+
+	filename := "test.py"
+	PyErr_SyntaxLocation(filename, 0)
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorSyntaxEx(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_SetNone(PyExc_SyntaxError)
+
+	filename := "test.py"
+	PyErr_SyntaxLocationEx(filename, 0, 0)
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorSyntaxLocation(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_SetNone(PyExc_SyntaxError)
+
+	filename := PyUnicode_FromString("test.py")
+	defer filename.DecRef()
+
+	PyErr_SyntaxLocationObject(filename, 0, 0)
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorExceptionMatches(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_SetNone(PyExc_BufferError)
+
+	assert.True(t, PyErr_ExceptionMatches(PyExc_BufferError))
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorGivenExceptionMatches(t *testing.T) {
+	Py_Initialize()
+
+	assert.True(t, PyErr_GivenExceptionMatches(PyExc_BufferError, PyExc_BufferError))
+}
+
+func TestErrorFetchRestore(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_SetNone(PyExc_BufferError)
+
+	exc, value, traceback := PyErr_Fetch()
+	assert.Nil(t, PyErr_Occurred())
+
+	assert.True(t, PyErr_GivenExceptionMatches(exc, PyExc_BufferError))
+	assert.Nil(t, value)
+	assert.Nil(t, traceback)
+
+	PyErr_Restore(exc, value, traceback)
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorNormalizeExceptionRestore(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_SetNone(PyExc_BufferError)
+
+	exc, value, traceback := PyErr_Fetch()
+	exc, value, traceback = PyErr_NormalizeException(exc, value, traceback)
+	assert.Nil(t, PyErr_Occurred())
+
+	assert.True(t, PyErr_GivenExceptionMatches(exc, PyExc_BufferError))
+	assert.Equal(t, 1, value.IsInstance(exc))
+	assert.Nil(t, traceback)
+
+	PyErr_Restore(exc, value, traceback)
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorGetSetExcInfo(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_SetNone(PyExc_BufferError)
+
+	exc, value, traceback := PyErr_GetExcInfo()
+
+	assert.True(t, PyErr_GivenExceptionMatches(exc, Py_None), PyUnicode_AsUTF8(exc.Repr()))
+	assert.Nil(t, value)
+	assert.Nil(t, traceback)
+
+	PyErr_SetExcInfo(exc, value, traceback)
+
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}
+
+func TestErrorInterrupt(t *testing.T) {
+	Py_Initialize()
+
+	PyErr_SetInterrupt()
+
+	assert.Equal(t, -1, PyErr_CheckSignals())
+
+	exc := PyErr_Occurred()
+	assert.True(t, PyErr_GivenExceptionMatches(exc, PyExc_TypeError))
+
+	assert.NotNil(t, PyErr_Occurred())
+	PyErr_Clear()
+	assert.Nil(t, PyErr_Occurred())
+}

exceptions.go

@@ -101,8 +101,8 @@ func PyException_GetTraceback(ex *PyObject) *PyObject {
 }
 
 //PyException_SetTraceback : https://docs.python.org/3/c-api/exceptions.html#c.PyException_SetTraceback
-func PyException_SetTraceback(ex, tb *PyObject) {
-	C.PyException_SetTraceback(toc(ex), toc(tb))
+func PyException_SetTraceback(ex, tb *PyObject) int {
+	return int(C.PyException_SetTraceback(toc(ex), toc(tb)))
 }
 
 //PyException_GetContext : https://docs.python.org/3/c-api/exceptions.html#c.PyException_GetContext

exceptions_test.go

@@ -0,0 +1,35 @@
+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))
+}

float_test.go

@@ -49,4 +49,12 @@ func TestPyFloatMinMax(t *testing.T) {
 	assert.Equal(t, math.MaxFloat64, PyFloat_GetMax())
 
 	assert.Equal(t, 2.2250738585072014e-308, PyFloat_GetMin())
+
+	PyFloat_ClearFreeList()
+}
+
+func TestPyFloatInfo(t *testing.T) {
+	Py_Initialize()
+
+	assert.NotNil(t, PyFloat_GetInfo())
 }

high_level_layer_test.go

@@ -0,0 +1,47 @@
+package python3
+
+import (
+	"io/ioutil"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRunFile(t *testing.T) {
+	Py_Initialize()
+
+	pyErr, err := PyRun_AnyFile("tests/test.py")
+	assert.Zero(t, pyErr)
+	assert.Nil(t, err)
+
+	stdout := PySys_GetObject("stdout")
+
+	result := stdout.CallMethodArgs("getvalue")
+	defer result.DecRef()
+
+	assert.Equal(t, "hello world\n", PyUnicode_AsUTF8(result))
+}
+
+func TestRunString(t *testing.T) {
+	Py_Initialize()
+
+	pythonCode, err := ioutil.ReadFile("tests/test.py")
+	assert.Nil(t, err)
+
+	assert.Zero(t, PyRun_SimpleString(string(pythonCode)))
+
+	stdout := PySys_GetObject("stdout")
+
+	result := stdout.CallMethodArgs("getvalue")
+	defer result.DecRef()
+
+	assert.Equal(t, "hello world\n", PyUnicode_AsUTF8(result))
+}
+
+func TestPyMain(t *testing.T) {
+	Py_Initialize()
+
+	pyErr, err := Py_Main([]string{"tests/test.py"})
+	assert.Zero(t, pyErr)
+	assert.Nil(t, err)
+}