tests: add module import example

sbinet · sbinet · commit 3c15159446b4 · 2015-12-21T10:15:45.000+01:00
Add a simple example on how to import and use `cPickle` from `go-python`. Updates sbinet#29.
diff --git a/tests/main.go b/tests/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+
 	"github.com/sbinet/go-python"
 )
 
@@ -17,4 +18,20 @@ func main() {
 	pystr := python.PyString_FromString(gostr)
 	str := python.PyString_AsString(pystr)
 	fmt.Println("hello [", str, "]")
+
+	pickle := python.PyImport_ImportModule("cPickle")
+	if pickle == nil {
+		panic("could not import 'cPickle'")
+	}
+	dumps := pickle.GetAttrString("dumps")
+	if dumps == nil {
+		panic("could not retrieve 'cPickle.dumps'")
+	}
+	out := dumps.CallFunctionObjArgs("O", pystr)
+	if out == nil {
+		panic("could not dump pystr")
+	}
+	fmt.Printf("cPickle.dumps(%s) = %q\n", gostr,
+		python.PyString_AsString(out),
+	)
 }
