def test_emit_operation(self):
opt = Optimizer()
opt.add_operation(Types.VOID, Operations.FINISH, [])
ops = opt.build_operations()
assert len(ops) == 1
assert ops[0].op == Operations.FINISH
assert ops[0].getarglist() == []
def test_simple_new(self, cpu):
opt = Optimizer([Virtualize])
struct_descr = cpu.new_struct()
opt.add_operation(Types.REF, Operations.NEW, [], descr=struct_descr)
ops = opt.build_operations()
assert len(ops) == 0
def test_subtraction(self):
opt = Optimizer([ConstantFold])
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
def test_cant_fold(self):
opt = Optimizer([ConstantFold])
i0 = opt.add_input(Types.INT)
opt.add_operation(Types.INT, Operations.INT_ADD,
[i0, opt.new_constant_int(1)]
)
ops = opt.build_operations()
assert len(ops) == 1
def test_inputs(self):
opt = Optimizer()
res = opt.add_input(Types.INT)
opt.add_operation(Types.VOID, Operations.FINISH, [res])
ops = opt.build_operations()
assert len(ops) == 1
assert ops[0].op == Operations.FINISH
assert ops[0].getarglist() == [res]
def test_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)
ops = opt.build_operations()
assert len(ops) == 0
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_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_setfield_virtual_on_nonvritual(self, cpu):
opt = Optimizer([Virtualize])
struct_descr = cpu.new_struct()
field_descr = cpu.new_field(struct_descr, Types.REF)
p0 = opt.add_input(Types.REF)
p1 = opt.add_operation(Operations.NEW, [], descr=struct_descr)
opt.add_operation(Operations.SETFIELD, [p0, p1], descr=field_descr)
ops = opt.build_operations()
assert len(ops) == 2
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_input(self, cpu):
opt = Optimizer([Virtualize])
struct_descr = cpu.new_struct()
field_descr = cpu.new_field(struct_descr, Types.INT)
i0 = opt.add_input(Types.INT)
p0 = opt.add_input(Types.REF)
opt.add_operation(Operations.SETFIELD, [p0, i0], descr=field_descr)
opt.add_operation(Operations.GETFIELD, [p0], descr=field_descr)
ops = opt.build_operations()
assert len(ops) == 2
def test_setfield_escapes(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)
opt.add_operation(Operations.FINISH, [p0])
ops = opt.build_operations()
assert len(ops) == 3
def test_gt(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(10)])
opt.add_operation(Types.VOID, Operations.GUARD_TRUE, [i1])
opt.add_operation(Types.INT, Operations.INT_GT,
[i0, opt.new_constant_int(5)])
ops = opt.build_operations()
assert len(ops) == 2
def test_lt(self):
opt = Optimizer([IntBounds, GuardPropagation])
i0 = opt.add_input(Types.INT)
i1 = opt.add_operation(Types.INT, Operations.INT_LT,
[i0, opt.new_constant_int(10)],
)
opt.add_operation(Types.VOID, Operations.GUARD_TRUE, [i1])
opt.add_operation(Types.INT, Operations.INT_LT, [i0, opt.new_constant_int(15)])
ops = opt.build_operations()
assert len(ops) == 2
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_cycle_of_virtuals(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.NEW, [], descr=struct_descr)
opt.add_operation(Operations.SETFIELD, [p0, p1], descr=field_descr)
opt.add_operation(Operations.SETFIELD, [p1, p0], descr=field_descr)
opt.add_operation(Operations.FINISH, [p0])
ops = opt.build_operations()
assert len(ops) == 5
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_addition(self):
opt = Optimizer([ConstantFold])
res = opt.add_operation(Types.INT, Operations.INT_ADD,
[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_multiple_setfields(self, cpu):
opt = Optimizer([Virtualize])
struct_descr = cpu.new_struct()
field_descr1 = cpu.new_field(struct_descr, Types.INT)
field_descr2 = cpu.new_field(struct_descr, Types.INT)
i0 = opt.add_input(Types.INT)
i1 = opt.add_input(Types.INT)
p0 = opt.add_operation(Operations.NEW, [], descr=struct_descr)
opt.add_operation(Operations.SETFIELD, [p0, i0], descr=field_descr1)
opt.add_operation(Operations.SETFIELD, [p0, i1], descr=field_descr2)
i2 = opt.add_operation(Operations.GETFIELD, [p0], descr=field_descr1)
i3 = opt.add_operation(Operations.GETFIELD, [p0], descr=field_descr2)
ops = opt.build_operations()
assert len(ops) == 0
assert opt.getvalue(i2) is i0
assert opt.getvalue(i3) is i1
def test_multiple_setfields(self, cpu):
opt = Optimizer([Virtualize])
struct_descr = cpu.new_struct()
field_descr1 = cpu.new_field(struct_descr)
field_descr2 = cpu.new_field(struct_descr)
i0 = opt.add_input(Types.INT)
i1 = opt.add_input(Types.INT)
p0 = opt.add_operation(Types.REF, Operations.NEW, [], descr=struct_descr)
opt.add_operation(Types.VOID, Operations.SETFIELD, [p0, i0], descr=field_descr1)
opt.add_operation(Types.VOID, Operations.SETFIELD, [p0, i1], descr=field_descr2)
i2 = opt.add_operation(Types.INT, Operations.GETFIELD, [p0], descr=field_descr1)
i3 = opt.add_operation(Types.INT, Operations.GETFIELD, [p0], descr=field_descr2)
ops = opt.build_operations()
assert len(ops) == 0
assert opt.getvalue(i2) is i0
assert opt.getvalue(i3) is i1
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
