def test_jle_doesnt_jump_when_greater_than():
vm = quaff.VirtualMachine()
vm.run(
[
quaff.inst_const(42),
quaff.inst_const(4),
quaff.inst_const(2),
quaff.inst_cmp(),
quaff.inst_jle(1),
quaff.inst_const(43),
]
)
assert_stack_elements(vm, [43, 42])
def test_jle_does_jump_when_equal():
vm = quaff.VirtualMachine()
vm.run(
[
quaff.inst_const(42),
quaff.inst_const(2),
quaff.inst_const(2),
quaff.inst_cmp(),
quaff.inst_jle(1),
quaff.inst_const(43),
quaff.inst_const(44),
]
)
assert_stack_elements(vm, [44, 42])
def test_can_calculate_factorial():
vm = quaff.VirtualMachine()
vm.run(
[
# n:
quaff.inst_const(6), # [6]
# n!:
quaff.inst_const(1), # [T1, 6]
quaff.inst_swap(1), # [6, T1]
quaff.inst_dup(), # [6, 6, T1]
quaff.inst_const(0), # [0, 6, 6, T1]
quaff.inst_cmp(), # [1, 6, T1]
quaff.inst_jle(7), # [6, T1]
quaff.inst_dup(), # [6, 6, T1]
quaff.inst_swap(2), # [T1, 6, 6]
quaff.inst_mul(), # [T6, 6]
quaff.inst_swap(1), # [6, T6]
quaff.inst_const(-1), # [-1, 6, T6]
quaff.inst_add(), # [5, T6]
quaff.inst_jmp(-11),
quaff.inst_pop(),
]
)
assert_stack_elements(vm, [720])
def test_jmp():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(1), quaff.inst_jmp(1), quaff.inst_const(2), quaff.inst_const(3)])
assert_stack_elements(vm, [3, 1])
def test_multiply_consts():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(7), quaff.inst_const(6), quaff.inst_mul()])
assert_stack_elements(vm, [42])
def test_add_consts():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(7), quaff.inst_const(42), quaff.inst_add()])
assert_stack_elements(vm, [49])
def test_cmp_ints_small_to_big():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(7), quaff.inst_const(42), quaff.inst_cmp()])
assert_stack_elements(vm, [-1])
def test_cmp_ints_equal():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(42), quaff.inst_const(42), quaff.inst_cmp()])
assert_stack_elements(vm, [0])
def test_const_then_swap_2():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(42), quaff.inst_const(7), quaff.inst_const(64), quaff.inst_swap(2)])
assert_stack_elements(vm, [42, 7, 64])
def test_const_1():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(1)])
assert_stack_elements(vm, [1])
def test_const_then_dup():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(42), quaff.inst_dup()])
assert_stack_elements(vm, [42, 42])
def test_lots_consts():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(i) for i in range(32)])
assert_equal(31, vm.read_stack_int32(0))
assert_equal(1, vm.read_stack_int32(30))
assert_equal(0, vm.read_stack_int32(31))
def test_multiple_consts():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(42), quaff.inst_const(64)])
assert_stack_elements(vm, [64, 42])
def test_negative_const():
vm = quaff.VirtualMachine()
vm.run([quaff.inst_const(-1)])
assert_stack_elements(vm, [-1])
