1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
import py
import re
from rpython.jit.metainterp import resoperation as rop
from rpython.jit.metainterp.history import AbstractDescr, AbstractFailDescr
def test_arity_mixins():
cases = [
(0, rop.NullaryOp),
(1, rop.UnaryOp),
(2, rop.BinaryOp),
(3, rop.TernaryOp),
(9, rop.N_aryOp)
]
def test_case(n, cls):
obj = cls()
obj.initarglist(range(n))
assert obj.getarglist() == range(n)
for i in range(n):
obj.setarg(i, i*2)
assert obj.numargs() == n
for i in range(n):
assert obj.getarg(i) == i*2
py.test.raises(IndexError, obj.getarg, n+1)
py.test.raises(IndexError, obj.setarg, n+1, 0)
for n, cls in cases:
test_case(n, cls)
def test_concrete_classes():
cls = rop.opclasses[rop.rop.INT_ADD]
assert issubclass(cls, rop.PlainResOp)
assert issubclass(cls, rop.BinaryOp)
assert cls.getopnum.im_func(cls) == rop.rop.INT_ADD
cls = rop.opclasses[rop.rop.CALL]
assert issubclass(cls, rop.ResOpWithDescr)
assert issubclass(cls, rop.N_aryOp)
assert cls.getopnum.im_func(cls) == rop.rop.CALL
cls = rop.opclasses[rop.rop.GUARD_TRUE]
assert issubclass(cls, rop.GuardResOp)
assert issubclass(cls, rop.UnaryOp)
assert cls.getopnum.im_func(cls) == rop.rop.GUARD_TRUE
def test_mixins_in_common_base():
INT_ADD = rop.opclasses[rop.rop.INT_ADD]
assert len(INT_ADD.__bases__) == 1
BinaryPlainResOp = INT_ADD.__bases__[0]
assert BinaryPlainResOp.__name__ == 'BinaryPlainResOp'
assert BinaryPlainResOp.__bases__ == (rop.BinaryOp, rop.PlainResOp)
INT_SUB = rop.opclasses[rop.rop.INT_SUB]
assert INT_SUB.__bases__[0] is BinaryPlainResOp
def test_instantiate():
op = rop.ResOperation(rop.rop.INT_ADD, ['a', 'b'], 'c')
assert op.getarglist() == ['a', 'b']
assert op.result == 'c'
assert repr(op) == "c = int_add(a, b)"
mydescr = AbstractDescr()
op = rop.ResOperation(rop.rop.CALL, ['a', 'b'], 'c', descr=mydescr)
assert op.getarglist() == ['a', 'b']
assert op.result == 'c'
assert op.getdescr() is mydescr
assert re.match("c = call\(a, b, descr=<.+>\)$", repr(op))
mydescr = AbstractFailDescr()
op = rop.ResOperation(rop.rop.GUARD_NO_EXCEPTION, [], None, descr=mydescr)
assert re.match("guard_no_exception\(descr=<.+>\)$", repr(op))
def test_can_malloc():
mydescr = AbstractDescr()
assert rop.ResOperation(rop.rop.NEW, [], 'b').can_malloc()
call = rop.ResOperation(rop.rop.CALL, ['a', 'b'], 'c', descr=mydescr)
assert call.can_malloc()
assert not rop.ResOperation(rop.rop.INT_ADD, ['a', 'b'], 'c').can_malloc()
def test_get_deep_immutable_oplist():
ops = [rop.ResOperation(rop.rop.INT_ADD, ['a', 'b'], 'c')]
newops = rop.get_deep_immutable_oplist(ops)
py.test.raises(TypeError, "newops.append('foobar')")
py.test.raises(TypeError, "newops[0] = 'foobar'")
py.test.raises(AssertionError, "newops[0].setarg(0, 'd')")
py.test.raises(AssertionError, "newops[0].setdescr('foobar')")
|
