[pypy-svn] r51081 - pypy/dist/pypy/jit/codegen/i386/test

[fijal at codespeak.net]

Author: fijal
Date: Sun Jan 27 13:38:02 2008
New Revision: 51081

Added:
   pypy/dist/pypy/jit/codegen/i386/test/test_regalloc.py   (contents, props changed)
Log:
Start real unittests for regalloc.


Added: pypy/dist/pypy/jit/codegen/i386/test/test_regalloc.py
==============================================================================
--- (empty file)
+++ pypy/dist/pypy/jit/codegen/i386/test/test_regalloc.py	Sun Jan 27 13:38:02 2008
@@ -0,0 +1,41 @@
+
+from pypy.jit.codegen.i386.regalloc  import RegAllocator
+from pypy.jit.codegen.i386 import operation as op
+from pypy.jit.codegen.i386.ri386 import mem, ebp, I386CodeBuilder
+from pypy.jit.codegen.model import GenVar
+
+WORD = op.WORD
+
+class DummyMc(I386CodeBuilder):
+    def __init__(self):
+        self.ops = []
+    
+    def __getattribute__(self, attr):
+        if attr == '__dict__' or attr == 'ops':
+            return object.__getattribute__(self, attr)
+        def append_to_ops(*args):
+            if attr == 'write':
+                return
+            self.ops.append((attr, args))
+        return append_to_ops
+        
+
+class TestRegalloc:
+    def test_basics(self):
+        """ This tests a simple two-arg routine
+        """
+        varx, vary = GenVar(), GenVar()
+        x = mem(ebp)
+        y = mem(ebp, WORD)
+        finalloc = mem(ebp, 2*WORD)
+        operations = [op.OpIntAdd(varx, vary)]
+        regalloc = RegAllocator(operations)
+        regalloc.set_final(operations, [finalloc])
+        regalloc.compute_lifetimes()
+        regalloc.init_reg_alloc([varx, vary], [x, y])
+        dummy_mc = DummyMc()
+        regalloc.generate_operations(dummy_mc)
+        regalloc.generate_final_moves(operations, [finalloc])
+        names = (op for op, args in dummy_mc.ops)
+        assert list(names) == ['MOV', 'ADD', 'PUSH', 'POP']
+        # XXX more asserts here