tests/modify-values: first import

sbinet · sbinet · commit 990e749f2e0b · 2016-08-05T10:18:49.000+02:00
diff --git a/python_test.go b/python_test.go
@@ -115,3 +115,18 @@ func TestErrFetch(t *testing.T) {
 		want: []byte("exc=&{<nil>}\nval=&{<nil>}\ntb=&{<nil>}\n"),
 	})
 }
+
+func TestModifyValues(t *testing.T) {
+	t.Parallel()
+	testPkg(t, pkg{
+		path: "tests/modify-values",
+		want: []byte(`values.__name__: "values"
+values.sval: "42"
+values.ival: 666
+sval='42'
+ival=666
+sval='42 is the answer'
+ival=1666
+`),
+	})
+}
diff --git a/tests/modify-values/main.go b/tests/modify-values/main.go
@@ -0,0 +1,76 @@
+package main
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/sbinet/go-python"
+)
+
+func init() {
+	err := python.Initialize()
+	if err != nil {
+		log.Panic(err.Error())
+	}
+}
+
+func main() {
+	module := python.PyImport_ImportModule("values")
+	if module == nil {
+		log.Fatal("could not import 'values'")
+	}
+
+	name := module.GetAttrString("__name__")
+	if name == nil {
+		log.Fatal("could not getattr '__name__'")
+	}
+	defer name.DecRef()
+	fmt.Printf("values.__name__: %q\n", python.PyString_AsString(name))
+
+	sval := module.GetAttrString("sval")
+	if sval == nil {
+		log.Fatal("could not getattr 'sval'")
+	}
+	defer sval.DecRef()
+	fmt.Printf("values.sval: %q\n", python.PyString_AsString(sval))
+
+	pyival := module.GetAttrString("ival")
+	if pyival == nil {
+		log.Fatal("could not getattr 'ival'")
+	}
+	defer pyival.DecRef()
+
+	ival := python.PyInt_AsLong(pyival)
+	fmt.Printf("values.ival: %d\n", ival)
+
+	myfunc := module.GetAttrString("myfunc")
+	if myfunc == nil {
+		log.Fatal("could not getattr 'myfunc'")
+	}
+	defer myfunc.DecRef()
+
+	o1 := myfunc.CallFunction()
+	if o1 == nil {
+		log.Fatal("could not call 'values.myfunc()'")
+	}
+	fmt.Printf("%s\n", python.PyString_AsString(o1))
+	o1.DecRef()
+
+	// modify 'test.ival' and 'test.sval'
+	{
+		pyival := python.PyInt_FromLong(ival + 1000)
+		module.SetAttrString("ival", pyival)
+		pyival.DecRef()
+
+		pysval := python.PyString_FromString(python.PyString_AsString(sval) + " is the answer")
+		module.SetAttrString("sval", pysval)
+		pysval.DecRef()
+	}
+
+	o2 := myfunc.CallFunction()
+	if o2 == nil {
+		log.Fatal("could not call 'values.myfunc()'")
+	}
+	fmt.Printf("%s\n", python.PyString_AsString(o2))
+	o2.DecRef()
+}
diff --git a/tests/modify-values/values.py b/tests/modify-values/values.py
@@ -0,0 +1,8 @@
+#!/usr/bin/env python2
+
+sval = "42"
+ival = 666
+
+def myfunc():
+    return "sval='%s'\nival=%d" %(sval,ival)
+
