working test

4 files changed, 111 insertions, 0 deletions

diff --git a/src/zope/tal/talpypy/zptest/__init__.py b/src/zope/tal/talpypy/zptest/__init__.py
new file mode 100644
index 0000000..b1c1f62
--- /dev/null
+++ b/src/zope/tal/talpypy/zptest/__init__.py
@@ -0,0 +1,13 @@
+from pypy.interpreter.mixedmodule import MixedModule 
+
+class Module(MixedModule):
+    """A ZPT module."""
+
+    interpleveldefs = {
+        'test'        : 'zpt.test',
+        'Interpreter' : 'zpt.Interpreter',
+    }
+
+    appleveldefs = {
+        'appfunc'     : 'appzpt.appfunc',
+    }
diff --git a/src/zope/tal/talpypy/zptest/appzpt.py b/src/zope/tal/talpypy/zptest/appzpt.py
new file mode 100644
index 0000000..6519dff
--- /dev/null
+++ b/src/zope/tal/talpypy/zptest/appzpt.py
@@ -0,0 +1,7 @@
+
+import zope.tal
+import random
+
+def appfunc():
+    return zope.tal.zptest.test(random.randrange(3))
+
diff --git a/src/zope/tal/talpypy/zptest/check.py b/src/zope/tal/talpypy/zptest/check.py
new file mode 100644
index 0000000..e0385dc
--- /dev/null
+++ b/src/zope/tal/talpypy/zptest/check.py
@@ -0,0 +1,18 @@
+import sys
+
+if len(sys.argv) == 2 and sys.argv[1] == 'so':
+    print 'so extension'
+    import zptest
+else:    
+    from pypy.interpreter.mixedmodule import testmodule
+    zptest =  testmodule('zptest', 'zope.tal.pypy')
+
+ip = zptest.Interpreter(None, 20)
+assert ip.power2() == 400
+assert ip.getInstance() is None
+
+ip = zptest.Interpreter([1, 2], 10)
+assert ip.power2() == 100
+assert ip.getInstance() == [1, 2, 3]
+
+print "tests passed"
diff --git a/src/zope/tal/talpypy/zptest/zpt.py b/src/zope/tal/talpypy/zptest/zpt.py
new file mode 100644
index 0000000..78c4453
--- /dev/null
+++ b/src/zope/tal/talpypy/zptest/zpt.py
@@ -0,0 +1,73 @@
+from pypy.interpreter.baseobjspace import ObjSpace, W_Root, Wrappable
+from pypy.interpreter.gateway import interp2app
+from pypy.interpreter.typedef import TypeDef, GetSetProperty
+
+def importModule(space, name):
+    w_builtin = space.getbuiltinmodule('__builtin__')
+    w_import = space.getattr(w_builtin, space.wrap('__import__'))
+    w_module = space.call_function(w_import, space.wrap(name))
+    return w_module
+
+def test(space, n):
+    return space.wrap(n + 1)
+test.unwrap_spec = [ObjSpace, int]
+
+class Interpreter(Wrappable):
+    def __init__(self, space, w_instance, code=1):
+        self.space = space
+        self.code = code
+        self.w_instance = w_instance
+
+    def multiply(self, w_y):
+        space = self.space
+        y = space.int_w(w_y)
+        return space.wrap(self.code * y)
+
+    def fget_code(space, self):
+        return space.wrap(self.code)
+
+    def fset_code(space, self, w_value):
+        self.code = space.int_w(w_value)
+
+    def call(self):
+        space = self.space
+        w_random = importModule(space, 'random')
+        w_randrange = space.getattr(w_random, space.wrap('randrange'))
+        rand = space.call_function(w_randrange, space.wrap(self.code))
+        return rand
+
+    def power2(self):
+        p = Power2(self.code)
+        return self.space.wrap(p.compute())
+
+    def getInstance(self):
+        space = self.space
+        if space.is_w(self.w_instance, space.wrap(None)):
+            return
+        w_append =space.getattr(self.w_instance, 'append')
+        space.call_function(w_append, space.wrap(3))
+        return self.w_instance
+
+class Power2:
+    def __init__(self, value):
+        self.value = value
+
+    def compute(self):
+        return self.value * self.value
+
+
+
+def interpreter_new(space, w_subtype, w_instance, x):
+    return space.wrap(Interpreter(space, w_instance, x))
+interpreter_new.unwrap_spec = [ObjSpace, W_Root, W_Root, int]
+
+getset_code = GetSetProperty(Interpreter.fget_code, Interpreter.fset_code, cls=Interpreter)
+
+Interpreter.typedef = TypeDef('Interpreter',
+    __new__ = interp2app(interpreter_new),
+    code = getset_code,
+    multiply = interp2app(Interpreter.multiply),
+    __call__ = interp2app(Interpreter.call),
+    power2 = interp2app(Interpreter.power2),
+    getInstance = interp2app(Interpreter.getInstance),
+)