def test_simple(self):
a = MyPPCAssembler()
a.blr()
f = make_func(a, "O", "O")
assert f(1) == 1
raises(TypeError, f)
raises(TypeError, f, 1, 2)
def access_at():
a = MyPPCAssembler()
a.lwzx(r3, r3, r4)
a.blr()
return make_func(a, "i", "ii")
def test_simple(self):
a = MyPPCAssembler()
a.blr()
f = make_func(a, "O", "O")
assert f(1) == 1
raises(TypeError, f)
raises(TypeError, f, 1, 2)
def test_floats(self):
a = MyPPCAssembler()
a.fadd(fr1, fr1, fr2)
a.blr()
f = make_func(a, 'f', 'ff')
raises(TypeError, f)
raises(TypeError, f, 1.0)
assert f(1.0, 2.0) == 3.0
raises(TypeError, f, 1.0, 2.0, 3.0)
raises(TypeError, f, 1.0, 2)
def test_signature(self):
a = MyPPCAssembler()
a.add(r3, r3, r4)
a.blr()
f = make_func(a, "i", "ii")
raises(TypeError, f)
raises(TypeError, f, 1)
assert f(1, 2) == 3
raises(TypeError, f, 1, 2, 3)
raises(TypeError, f, 1, "2")
def test_fast_entry(self):
a = MyPPCAssembler()
a.blr()
f = make_func(a, "O", "O")
assert f(1) == 1
b = MyPPCAssembler()
from pypy.jit.codegen.ppc.ppcgen import util
# eurgh!:
b.load_word(r0, util.access_at(id(f.code), 8) + f.FAST_ENTRY_LABEL)
b.mtctr(r0)
b.bctr()
g = make_func(b, "O", "O")
assert g(1) == 1
def test_less_simple(self):
a = MyPPCAssembler()
s = lookup("PyNumber_Add")
a.load_word(r5, s)
a.mtctr(r5)
a.bctr()
f = make_func(a, "O", "OO")
raises(TypeError, f)
raises(TypeError, f, 1)
assert f(1, 2) == 3
raises(TypeError, f, 1, 2, 3)
def access_at():
a = MyPPCAssembler()
a.lwzx(r3, r3, r4)
a.blr()
return make_func(a, "i", "ii")
def test_less_simple(self):
a = MyPPCAssembler()
s = lookup("PyNumber_Add")
a.load_word(r5, s)
a.mtctr(r5)
a.bctr()
f = make_func(a, "O", "OO")
raises(TypeError, f)
raises(TypeError, f, 1)
assert f(1, 2) == 3
raises(TypeError, f, 1, 2, 3)
def test_signature(self):
a = MyPPCAssembler()
a.add(r3, r3, r4)
a.blr()
f = make_func(a, "i", "ii")
raises(TypeError, f)
raises(TypeError, f, 1)
assert f(1, 2) == 3
raises(TypeError, f, 1, 2, 3)
raises(TypeError, f, 1, "2")
def test_floats(self):
a = MyPPCAssembler()
a.fadd(fr1, fr1, fr2)
a.blr()
f = make_func(a, 'f', 'ff')
raises(TypeError, f)
raises(TypeError, f, 1.0)
assert f(1.0, 2.0) == 3.0
raises(TypeError, f, 1.0, 2.0, 3.0)
raises(TypeError, f, 1.0, 2)
def test_raise(self):
a = MyPPCAssembler()
a.mflr(0)
a.stw(0, 1, 8)
a.stwu(1, 1, -80)
err_set = lookup("PyErr_SetObject")
exc = lookup("PyExc_ValueError")
a.load_word(5, err_set)
a.mtctr(5)
a.load_from(3, exc)
a.mr(4, 3)
a.bctrl()
a.li(3, 0)
a.lwz(0, 1, 88)
a.addi(1, 1, 80)
a.mtlr(0)
a.blr()
raises(ValueError, a.assemble())
def test_intcheck(self):
a = MyPPCAssembler()
a.lwz(r5, r4, pystructs.PyVarObject.ob_size)
a.cmpwi(r5, 1)
a.bne("not_one")
a.lwz(r5, r4, pystructs.PyTupleObject.ob_item + 0*4)
a.lwz(r5, r5, 4)
a.load_word(r6, lookup("PyInt_Type"))
a.cmpw(r5, r6)
a.bne("not_int")
a.li(r3, 1)
a.b("exit")
a.label("not_int")
a.li(r3, 0)
a.b("exit")
a.label("not_one")
a.li(r3, 2)
a.label("exit")
a.load_word(r5, lookup("PyInt_FromLong"))
a.mtctr(r5)
a.bctr()
f = a.assemble()
assert f() == 2
assert f("", "") == 2
assert f("") == 0
assert f(1) == 1
def test_tuplelength2(self):
a = MyPPCAssembler()
a.mflr(0)
a.stw(0, 1, 8)
a.stwu(1, 1, -80)
a.mr(3, 4)
a.load_word(5, lookup("PyTuple_Size"))
a.mtctr(5)
a.bctrl()
a.load_word(5, lookup("PyInt_FromLong"))
a.mtctr(5)
a.bctrl()
a.lwz(0, 1, 88)
a.addi(1, 1, 80)
a.mtlr(0)
a.blr()
f = a.assemble()
assert f() == 0
assert f(1) == 1
assert f('') == 1
assert f('', 3) == 2
def test_tuplelength(self):
a = MyPPCAssembler()
a.lwz(3, 4, pystructs.PyVarObject.ob_size)
a.load_word(5, lookup("PyInt_FromLong"))
a.mtctr(5)
a.bctr()
f = a.assemble()
assert f() == 0
assert f(1) == 1
assert f('') == 1
def testSucceed(idesc, *args):
a = MyPPCAssembler()
# "assertNotRaises" :-)
idesc.__get__(a)(*args)
def testFailure(idesc, *args):
a = MyPPCAssembler()
raises(ValueError, idesc.__get__(a), *args)
def test_numberadd(self):
a = MyPPCAssembler()
a.lwz(r5, r4, pystructs.PyVarObject.ob_size)
a.cmpwi(r5, 2)
a.bne("err_out")
a.lwz(r3, r4, 12)
a.lwz(r4, r4, 16)
a.load_word(r5, lookup("PyNumber_Add"))
a.mtctr(r5)
a.bctr()
a.label("err_out")
a.mflr(r0)
a.stw(r0, r1, 8)
a.stwu(r1, r1, -80)
err_set = lookup("PyErr_SetObject")
exc = lookup("PyExc_TypeError")
a.load_word(r5, err_set)
a.mtctr(r5)
a.load_from(r3, exc)
a.mr(r4, r3)
a.bctrl()
a.li(r3, 0)
a.lwz(r0, r1, 88)
a.addi(r1, r1, 80)
a.mtlr(r0)
a.blr()
f = a.assemble()
raises(TypeError, f)
raises(TypeError, f, '', 1)
raises(TypeError, f, 1)
raises(TypeError, f, 1, 2, 3)
assert f(1, 2) == 3
assert f('a', 'b') == 'ab'
def test_makestring(self):
a = MyPPCAssembler()
a.li(r3, 0)
a.li(r4, 0)
a.load_word(r5, lookup("PyString_FromStringAndSize"))
a.mtctr(r5)
a.bctr()
f = a.assemble()
assert f() == ''
def test_signature2(self):
a = MyPPCAssembler()
a.add(r3, r3, r4)
a.add(r3, r3, r5)
a.add(r3, r3, r6)
a.add(r3, r3, r7)
s = lookup("PyInt_FromLong")
a.load_word(r0, s)
a.mtctr(r0)
a.bctr()
f = make_func(a, "O", "iiiii")
raises(TypeError, f)
raises(TypeError, f, 1)
assert f(1, 2, 3, 4, 5) == 1 + 2 + 3 + 4 + 5
raises(TypeError, f, 1, 2, 3)
raises(TypeError, f, 1, "2", 3, 4, 5)
def itoO():
a = MyPPCAssembler()
a.blr()
return make_func(a, "O", "i")
def test_fast_entry(self):
a = MyPPCAssembler()
a.blr()
f = make_func(a, "O", "O")
assert f(1) == 1
b = MyPPCAssembler()
from pypy.jit.codegen.ppc.ppcgen import util
# eurgh!:
b.load_word(r0, util.access_at(id(f.code), 8) + f.FAST_ENTRY_LABEL)
b.mtctr(r0)
b.bctr()
g = make_func(b, "O", "O")
assert g(1) == 1
def wrap(funcname, retcode, signature):
argcount = len(signature)
ourcode = MyPPCAssembler()
ourcode.mflr(r0)
ourcode.stmw(r31, r1, -4)
ourcode.stw(r0, r1, 8)
ourcode.stwu(r1, r1, -80)
ourcode.lwz(r3, r4, 8)
ourcode.cmpwi(r3, argcount)
ourcode.bne("argserror")
assert argcount < 9
if argcount > 0:
load_arg(ourcode, 0, signature[0])
for i in range(2, argcount):
load_arg(ourcode, i, signature[i])
if argcount > 1:
load_arg(ourcode, 1, signature[1])
ourcode.load_word(r0, lookup(funcname))
ourcode.mtctr(r0)
ourcode.bctrl()
if retcode == 'i':
s = lookup("PyInt_FromLong")
ourcode.load_word(r0, s)
ourcode.mtctr(r0)
ourcode.bctrl()
elif retcode == 'f':
s = lookup("PyFloat_FromDouble")
ourcode.load_word(r0, s)
ourcode.mtctr(r0)
ourcode.bctrl()
ourcode.label("epilogue")
ourcode.lwz(r0, r1, 88)
ourcode.addi(r1, r1, 80)
ourcode.mtlr(r0)
ourcode.lmw(r31, r1, -4)
ourcode.blr()
err_set = lookup("PyErr_SetObject")
exc = lookup("PyExc_TypeError")
ourcode.label("argserror")
ourcode.load_word(r5, err_set)
ourcode.mtctr(r5)
ourcode.load_from(r3, exc)
ourcode.mr(r4, r3)
ourcode.bctrl()
ourcode.li(r3, 0)
ourcode.b("epilogue")
return ourcode.assemble()
def make_func(code, retcode, signature, localwords=0):
"""code shouldn't contain prologue/epilogue (or touch r31)"""
stacksize = 80 + 4*localwords
argcount = len(signature)
ourcode = MyPPCAssembler()
ourcode.mflr(r0)
ourcode.stmw(r31, r1, -4)
ourcode.stw(r0, r1, 8)
ourcode.stwu(r1, r1, -stacksize)
ourcode.lwz(r3, r4, 8)
ourcode.cmpwi(r3, argcount)
ourcode.bne("argserror")
assert argcount < 9
if argcount > 0:
load_arg(ourcode, 0, signature[0])
for i in range(2, argcount):
load_arg(ourcode, i, signature[i])
if argcount > 1:
load_arg(ourcode, 1, signature[1])
ourcode.bl(FAST_ENTRY_LABEL)
if retcode == 'i':
s = lookup("PyInt_FromLong")
ourcode.load_word(r0, s)
ourcode.mtctr(r0)
ourcode.bctrl()
elif retcode == 'f':
s = lookup("PyFloat_FromDouble")
ourcode.load_word(r0, s)
ourcode.mtctr(r0)
ourcode.bctrl()
ourcode.label("epilogue")
ourcode.lwz(r0, r1, stacksize + 8)
ourcode.addi(r1, r1, stacksize)
ourcode.mtlr(r0)
ourcode.lmw(r31, r1, -4)
ourcode.blr()
err_set = lookup("PyErr_SetObject")
exc = lookup("PyExc_TypeError")
ourcode.label("argserror")
ourcode.load_word(r5, err_set)
ourcode.mtctr(r5)
ourcode.load_from(r3, exc)
ourcode.mr(r4, r3)
ourcode.bctrl()
ourcode.li(r3, 0)
ourcode.b("epilogue")
ourcode.label(FAST_ENTRY_LABEL)
# err, should be an Assembler method:
l = {}
for k in code.labels:
l[k] = code.labels[k] + 4*len(ourcode.insts)
r = code.rlabels.copy()
for k in code.rlabels:
r[k + 4*len(ourcode.insts)] = code.rlabels[k]
ourcode.insts.extend(code.insts)
ourcode.labels.update(l)
ourcode.rlabels.update(r)
r = ourcode.assemble()
r.FAST_ENTRY_LABEL = ourcode.labels[FAST_ENTRY_LABEL]
return r
def test_signature2(self):
a = MyPPCAssembler()
a.add(r3, r3, r4)
a.add(r3, r3, r5)
a.add(r3, r3, r6)
a.add(r3, r3, r7)
s = lookup("PyInt_FromLong")
a.load_word(r0, s)
a.mtctr(r0)
a.bctr()
f = make_func(a, "O", "iiiii")
raises(TypeError, f)
raises(TypeError, f, 1)
assert f(1, 2, 3, 4, 5) == 1 + 2 + 3 + 4 + 5
raises(TypeError, f, 1, 2, 3)
raises(TypeError, f, 1, "2", 3, 4, 5)
def itoO():
a = MyPPCAssembler()
a.blr()
return make_func(a, "O", "i")
