[pypy-commit] pypy generator-in-rpython: A first version of the test.

[arigo]

Author: Armin Rigo <arigo at tunes.org>
Branch: generator-in-rpython
Changeset: r49967:026ff0f93709
Date: 2011-11-29 15:58 +0100
http://bitbucket.org/pypy/pypy/changeset/026ff0f93709/

Log:	A first version of the test.

diff --git a/pypy/translator/test/test_generator.py b/pypy/translator/test/test_generator.py
new file mode 100644
--- /dev/null
+++ b/pypy/translator/test/test_generator.py
@@ -0,0 +1,126 @@
+from pypy.objspace.flow.objspace import FlowObjSpace
+from pypy.translator.translator import TranslationContext
+from pypy.translator.generator import replace_graph_with_bootstrap
+
+
+# ____________________________________________________________
+
+def f_gen(n):
+    i = 0
+    while i < n:
+        yield i
+        i += 1
+
+class GeneratorIterator(object):
+    def __init__(self, entry):
+        self.current = entry
+    def next(self):
+        e = self.current
+        self.current = None
+        if isinstance(e, Yield1):
+            n = e.n_0
+            i = e.i_0
+            i += 1
+        else:
+            n = e.n_0
+            i = 0
+        if i < n:
+            e = Yield1()
+            e.n_0 = n
+            e.i_0 = i
+            self.current = e
+            return i
+        raise StopIteration
+
+    def __iter__(self):
+        return self
+
+class AbstractPosition(object):
+    _immutable_ = True
+class Entry1(AbstractPosition):
+    _immutable_ = True
+class Yield1(AbstractPosition):
+    _immutable_ = True
+
+def f_explicit(n):
+    e = Entry1()
+    e.n_0 = n
+    return GeneratorIterator(e)
+
+def test_explicit():
+    assert list(f_gen(10)) == list(f_explicit(10))
+
+# ____________________________________________________________
+
+
+class TestGenerator:
+
+    def test_replace_graph_with_bootstrap(self):
+        def func(n, x, y, z):
+            yield n
+            yield n
+        #
+        space = FlowObjSpace()
+        graph = space.build_flow(func)
+        assert graph.startblock.operations[0].opname == 'generator_entry'
+        replace_graph_with_bootstrap(graph, 'newgraph')
+        block = graph.startblock
+        ops = block.operations
+        assert ops[0].opname == 'call'      # e = Entry1()
+        assert ops[1].opname == 'setattr'   # e.n_0 = n
+        assert ops[1].args[1].value.startswith('n_')
+        assert ops[2].opname == 'setattr'   # e.x_0 = x
+        assert ops[2].args[1].value.startswith('x_')
+        assert ops[3].opname == 'setattr'   # e.y_0 = y
+        assert ops[3].args[1].value.startswith('y_')
+        assert ops[4].opname == 'setattr'   # e.z_0 = z
+        assert ops[4].args[1].value.startswith('z_')
+        assert ops[5].opname == 'call'      # g = Generator(e)
+        assert ops[5].args[1] == ops[0].result
+        assert len(ops) == 6
+        assert len(block.exits) == 1
+        assert block.exits[0].target is graph.returnblock
+
+    def test_make_generator_body_graph(self):
+        def f(n, x, y, z):
+            z *= 10
+            yield n
+            z -= 10
+        #
+        def f__next(generator):
+            n = generator.n_0
+            x = generator.x_0
+            y = generator.y_0
+            z = generator.z_0
+            e = generator.current
+            generator.current = None
+            if isinstance(e, "some class"):
+                xxx
+        #
+        space = FlowObjSpace()
+        graph = space.build_flow(func)
+        newgraph = make_generator_body_graph(graph)
+        assert len(newgraph.startblock.inputargs) == 1
+        [v_generator] = newgraph.startblock.inputargs
+        ops = newgraph.startblock.operations
+        assert ops[0].opname == 'getattr'   # n = g.n_0
+        assert ops[0].args[0] == v_generator
+        assert ops[0].args[1].value.startswith('n_')
+        assert ops[1].opname == 'getattr'   # x = g.x_0
+        assert ops[1].args[0] == v_generator
+        assert ops[1].args[1].value.startswith('x_')
+        assert ops[2].opname == 'getattr'   # y = g.y_0
+        assert ops[2].args[0] == v_generator
+        assert ops[2].args[1].value.startswith('y_')
+        assert ops[3].opname == 'getattr'   # z = g.z_0
+        assert ops[3].args[0] == v_generator
+        assert ops[3].args[1].value.startswith('z_')
+        assert ops[4].opname == 'getattr'   # e = g.current
+        assert ops[4].args[0] == v_generator
+        assert ops[4].args[1].value == 'current'
+        assert ops[5].opname == 'setattr'   # g.current = None
+        assert ops[5].args[0] == v_generator
+        assert ops[5].args[1].value == 'current'
+        assert ops[6].opname == 'call'      # isinstance(e, Yield1)
+        assert ops[6].args[0].value == isinstance
+        assert len(ops) == 7