def test_subtraction(self):
opt = Optimizer([ConstantFold])
res = opt.add_operation(Types.INT, Operations.INT_SUB,
[opt.new_constant_int(1), opt.new_constant_int(0)]
)
ops = opt.build_operations()
assert len(ops) == 0
assert opt.getvalue(res).getint() == 1
def test_subtraction(self):
opt = Optimizer([ConstantFold])
res = opt.add_operation(
Operations.INT_SUB,
[opt.new_constant_int(1),
opt.new_constant_int(0)])
ops = opt.build_operations()
assert len(ops) == 0
assert opt.getvalue(res).getint() == 1
def test_known_guard_true(self):
opt = Optimizer([GuardPropagation])
i0 = opt.add_input(Types.INT)
opt.add_operation(Types.VOID, Operations.GUARD_TRUE, [i0])
opt.add_operation(Types.VOID, Operations.GUARD_TRUE, [i0])
ops = opt.build_operations()
assert len(ops) == 1
assert opt.getvalue(i0).getint() == 1
def test_guard_false(self):
opt = Optimizer([ConstantFold, GuardPropagation])
i0 = opt.add_input(Types.INT)
opt.add_operation(Types.VOID, Operations.GUARD_FALSE, [i0])
opt.add_operation(Types.INT, Operations.INT_EQ, [i0, opt.new_constant_int(1)])
ops = opt.build_operations()
assert len(ops) == 1
assert opt.getvalue(i0).getint() == 0
def test_guard_value(self):
opt = Optimizer([GuardPropagation])
i0 = opt.add_input(Types.INT)
opt.add_operation(Operations.GUARD_VALUE,
[i0, opt.new_constant_int(5)])
ops = opt.build_operations()
assert len(ops) == 1
assert opt.getvalue(i0).getint() == 5
def test_known_guard_true(self):
opt = Optimizer([GuardPropagation])
i0 = opt.add_input(Types.INT)
opt.add_operation(Operations.GUARD_TRUE, [i0])
opt.add_operation(Operations.GUARD_TRUE, [i0])
ops = opt.build_operations()
assert len(ops) == 1
assert opt.getvalue(i0).getint() == 1
def test_guard_false(self):
opt = Optimizer([ConstantFold, GuardPropagation])
i0 = opt.add_input(Types.INT)
opt.add_operation(Operations.GUARD_FALSE, [i0])
opt.add_operation(Operations.INT_EQ, [i0, opt.new_constant_int(1)])
ops = opt.build_operations()
assert len(ops) == 1
assert opt.getvalue(i0).getint() == 0
def test_read_unsetfield(self, cpu):
opt = Optimizer([Virtualize])
struct_descr = cpu.new_struct()
field_descr = cpu.new_field(struct_descr)
p0 = opt.add_operation(Types.REF, Operations.NEW, [], descr=struct_descr)
i0 = opt.add_operation(Types.INT, Operations.GETFIELD, [p0], descr=field_descr)
ops = opt.build_operations()
assert len(ops) == 0
assert opt.getvalue(i0).getint() == 0
def test_read_unsetfield_ref(self, cpu):
opt = Optimizer([Virtualize])
struct_descr = cpu.new_struct()
field_descr = cpu.new_field(struct_descr, Types.REF)
p0 = opt.add_operation(Operations.NEW, [], descr=struct_descr)
p1 = opt.add_operation(Operations.GETFIELD, [p0], descr=field_descr)
ops = opt.build_operations()
assert len(ops) == 0
assert opt.getvalue(p1).getref() is None
def test_read_unsetfield_float(self, cpu):
opt = Optimizer([Virtualize])
struct_descr = cpu.new_struct()
field_descr = cpu.new_field(struct_descr, Types.FLOAT)
p0 = opt.add_operation(Operations.NEW, [], descr=struct_descr)
f0 = opt.add_operation(Operations.GETFIELD, [p0], descr=field_descr)
ops = opt.build_operations()
assert len(ops) == 0
assert opt.getvalue(f0).getfloat() == 0.0
def test_get_setfield(self, cpu):
opt = Optimizer([Virtualize])
i0 = opt.add_input(Types.INT)
struct_descr = cpu.new_struct()
field_descr = cpu.new_field(struct_descr)
p0 = opt.add_operation(Types.REF, Operations.NEW, [], descr=struct_descr)
opt.add_operation(Types.VOID, Operations.SETFIELD, [p0, i0], descr=field_descr)
i1 = opt.add_operation(Types.INT, Operations.GETFIELD, [p0], descr=field_descr)
ops = opt.build_operations()
assert len(ops) == 0
assert opt.getvalue(i1) is i0
def test_get_setfield(self, cpu):
opt = Optimizer([Virtualize])
i0 = opt.add_input(Types.INT)
struct_descr = cpu.new_struct()
field_descr = cpu.new_field(struct_descr, Types.INT)
p0 = opt.add_operation(Operations.NEW, [], descr=struct_descr)
opt.add_operation(Operations.SETFIELD, [p0, i0], descr=field_descr)
i1 = opt.add_operation(Operations.GETFIELD, [p0], descr=field_descr)
ops = opt.build_operations()
assert len(ops) == 0
assert opt.getvalue(i1) is i0
def test_ge_reverse(self):
opt = Optimizer([IntBounds, GuardPropagation])
i0 = opt.add_input(Types.INT)
i1 = opt.add_operation(Operations.INT_LT,
[i0, opt.new_constant_int(5)])
opt.add_operation(Operations.GUARD_TRUE, [i1])
i2 = opt.add_operation(Operations.INT_GE,
[i0, opt.new_constant_int(7)])
opt.add_operation(Operations.GUARD_FALSE, [i2])
ops = opt.build_operations()
assert len(ops) == 2
assert opt.getvalue(i2).getint() == 0
def test_lt_reverse(self):
opt = Optimizer([IntBounds, GuardPropagation])
i0 = opt.add_input(Types.INT)
i1 = opt.add_operation(Types.INT, Operations.INT_GT,
[i0, opt.new_constant_int(5)]
)
opt.add_operation(Types.VOID, Operations.GUARD_TRUE, [i1])
i2 = opt.add_operation(Types.INT, Operations.INT_LT,
[i0, opt.new_constant_int(3)]
)
opt.add_operation(Types.VOID, Operations.GUARD_FALSE, [i2])
ops = opt.build_operations()
assert len(ops) == 2
assert opt.getvalue(i2).getint() == 0
