def test_binop():
for op, bc in [
('+', BINARY_ADD),
('==', BINARY_EQ),
('<', BINARY_LT),
('-', BINARY_SUB),
('*', BINARY_MUL),
('/', BINARY_DIV),
('%', BINARY_MOD),
]:
bytecode = compile_ast(BinOp(op, Variable('x'), Variable('y')))
assert bytecode.code == to_code([
LOAD_VAR, 0,
LOAD_VAR, 1,
bc, 0,
RETURN, 0])
assert bytecode.names == ['x', 'y']
assert bytecode.constants_float == []
bytecode = compile_ast(BinOp(op, ConstantNum(2.0), Variable('y')))
assert bytecode.code == to_code([
LOAD_CONSTANT_FLOAT, 0,
LOAD_VAR, 0,
bc, 0,
RETURN, 0])
assert bytecode.names == ['y']
assert bytecode.constants_float == [2.0]
def test_variable():
bytecode = compile_ast(Variable('x'))
assert bytecode.code == to_code([LOAD_VAR, 0, RETURN, 0])
assert bytecode.names == ['x']
assert bytecode.constants_float == []
bytecode = compile_ast(Block([
Variable('zzz'), Variable('y'), Variable('zzz')]))
assert bytecode.code == to_code([
LOAD_VAR, 0, LOAD_VAR, 1, LOAD_VAR, 0, RETURN, 0])
assert bytecode.names == ['zzz', 'y']
assert bytecode.constants_float == []
def test_assignment():
bytecode = compile_ast(Assignment('x', Variable('y')))
assert bytecode.code == to_code([LOAD_VAR, 0, ASSIGN, 1, RETURN, 0])
assert bytecode.names == ['y', 'x']
assert bytecode.constants_float == []
bytecode = compile_ast(Assignment('x', ConstantNum(13.4)))
assert bytecode.code == to_code([
LOAD_CONSTANT_FLOAT, 0,
ASSIGN, 0,
RETURN, 0])
assert bytecode.names == ['x']
assert bytecode.constants_float == [13.4]
def test_load_constant():
bc = ByteCode(to_code([
LOAD_CONSTANT_FLOAT, 0,
RETURN, 0]),
[], [12.2], [], [])
frame = interpret(bc)
assert frame.test_valuestack == [W_FloatObject(12.2)]
assert frame.vars == []
def test_block():
bytecode = compile_ast(Block([
Stmt(Variable('xyz')), Stmt(ConstantNum(12.3))]))
assert bytecode.code == to_code([
LOAD_VAR, 0, DISCARD_TOP, 0,
LOAD_CONSTANT_FLOAT, 0, DISCARD_TOP, 0,
RETURN, 0])
assert bytecode.names == ['xyz']
assert bytecode.constants_float == [12.3]
def test_get_location():
filename = '_for_test_jit.js'
source = '''
function foo(x, y) {
return x + y;
};
x = 3;
foo(1, x);
'''
ast = parser.parse(source, filename=filename)
bc = CompilerContext.compile_ast(ast, co_filename=filename)
assert bc.code == to_code([
LOAD_CONSTANT_FN, 0,
ASSIGN, 0,
LOAD_CONSTANT_FN, 0,
DISCARD_TOP, 0,
LOAD_CONSTANT_FLOAT, 0,
ASSIGN, 1,
LOAD_VAR, 0,
LOAD_CONSTANT_FLOAT, 1,
LOAD_VAR, 1,
CALL, 2,
DISCARD_TOP, 0,
RETURN, 0])
assert bc.co_name == '__main__'
assert bc.co_filename == '_for_test_jit.js'
assert bc.co_firstlineno == 0
inner_bc = bc.constants_fn[0]
assert inner_bc.code == to_code([
LOAD_VAR, 0,
LOAD_VAR, 1,
BINARY_ADD, 0,
RETURN, 1])
assert inner_bc.co_name == 'foo'
assert inner_bc.co_filename == '_for_test_jit.js'
assert inner_bc.co_firstlineno == 1
assert get_printable_location(0, bc.code, bc) == \
"<code object __main__, file '_for_test_jit.js', line 1> #0 LOAD_CONSTANT_FN"
assert get_printable_location(2, bc.code, bc) == \
"<code object __main__, file '_for_test_jit.js', line 1> #2 ASSIGN"
assert get_printable_location(0, inner_bc.code, inner_bc) == \
"<code object foo, file '_for_test_jit.js', line 2> #0 LOAD_VAR"
assert get_printable_location(4, inner_bc.code, inner_bc) == \
"<code object foo, file '_for_test_jit.js', line 2> #4 BINARY_ADD"
def test_call():
bytecode = compile_ast(Call(Variable('fn'), []))
assert bytecode.code == to_code([
LOAD_VAR, 0,
CALL, 0,
RETURN, 0])
assert bytecode.names == ['fn']
assert bytecode.constants_float == []
bytecode = compile_ast(Call(Variable('fn'), [ConstantNum(1.0)]))
assert bytecode.code == to_code([
LOAD_VAR, 0,
LOAD_CONSTANT_FLOAT, 0,
CALL, 1,
RETURN, 0])
assert bytecode.names == ['fn']
assert bytecode.constants_float == [1.0]
bytecode = compile_ast(Call(Variable('fn'), [
Variable('z'), ConstantNum(1.0)]))
assert bytecode.code == to_code([
LOAD_VAR, 0,
LOAD_VAR, 1,
LOAD_CONSTANT_FLOAT, 0,
CALL, 2,
RETURN, 0])
assert bytecode.names == ['fn', 'z']
assert bytecode.constants_float == [1.0]
bytecode = compile_ast(Call(Variable('fn'), [
Call(Variable('z'), []), ConstantNum(1.0)]))
assert bytecode.code == to_code([
LOAD_VAR, 0,
LOAD_VAR, 1,
CALL, 0,
LOAD_CONSTANT_FLOAT, 0,
CALL, 2,
RETURN, 0])
assert bytecode.names == ['fn', 'z']
assert bytecode.constants_float == [1.0]
def test_jumps():
bc = ByteCode(to_code([
LOAD_CONSTANT_FLOAT, 0,
JUMP_IF_FALSE, 8,
LOAD_CONSTANT_FLOAT, 1,
JUMP_ABSOLUTE, 10,
LOAD_CONSTANT_FLOAT, 2,
RETURN, 0]),
[], [0.0, -1.0, 1.0], [], [])
frame = interpret(bc)
assert frame.test_valuestack == [W_FloatObject(1.0)]
bc = ByteCode(to_code([
LOAD_CONSTANT_FLOAT, 0,
JUMP_IF_FALSE, 8,
LOAD_CONSTANT_FLOAT, 1,
JUMP_ABSOLUTE, 10,
LOAD_CONSTANT_FLOAT, 2,
RETURN, 0]),
[], [1.0, -1.0, 2.0], [], [])
frame = interpret(bc)
assert frame.test_valuestack == [W_FloatObject(-1.0)]
def test_while():
bytecode = compile_ast(Block([
Variable('x'),
While(ConstantNum(1.0), ConstantNum(1.0))]))
expected_code = to_code([
LOAD_VAR, 0,
LOAD_CONSTANT_FLOAT, 0,
JUMP_IF_FALSE, 10,
LOAD_CONSTANT_FLOAT, 1,
JUMP_ABSOLUTE, 2,
RETURN, 0])
assert bytecode.code == expected_code
assert bytecode.names == ['x']
assert bytecode.constants_float == [1.0, 1.0]
def test_if():
bytecode = compile_ast(If(ConstantNum(1.0), ConstantNum(2.0)))
expected_code = to_code([
LOAD_CONSTANT_FLOAT, 0,
JUMP_IF_FALSE, 6,
LOAD_CONSTANT_FLOAT, 1,
RETURN, 0])
assert bytecode.code == expected_code
assert bytecode.names == []
assert bytecode.constants_float == [1.0, 2.0]
bytecode = compile_ast(
If(ConstantNum(1.0), ConstantNum(2.0), ConstantNum(3.0)))
expected_code = to_code([
LOAD_CONSTANT_FLOAT, 0,
JUMP_IF_FALSE, 8,
LOAD_CONSTANT_FLOAT, 1,
JUMP_ABSOLUTE, 10,
LOAD_CONSTANT_FLOAT, 2,
RETURN, 0])
print dis(bytecode.code)
assert bytecode.code == expected_code
assert bytecode.names == []
assert bytecode.constants_float == [1.0, 2.0, 3.0]
def test_return():
bytecode = compile_ast(FnDef('foo', ['x'], Block([
Return(Variable('x'))]), None, 0))
expected_code = to_code([
LOAD_CONSTANT_FN, 0,
ASSIGN, 0,
LOAD_CONSTANT_FN, 0,
RETURN, 0])
expected_inner_bytecode = to_code([
LOAD_VAR, 0,
RETURN, 1
])
bytecode = compile_ast(FnDef('foo', ['x'], Block([
Return()]), None, 0))
expected_code = to_code([
LOAD_CONSTANT_FN, 0,
ASSIGN, 0,
LOAD_CONSTANT_FN, 0,
RETURN, 0])
expected_inner_bytecode = to_code([RETURN, 0])
assert bytecode.code == expected_code
assert bytecode.names == ['foo']
assert len(bytecode.constants_fn) == 1
inner = bytecode.constants_fn[0]
assert inner.code == expected_inner_bytecode
assert inner.names == ['x']
assert inner.constants_float == []
bytecode = compile_ast(FnDef('foo', ['x'], Block([
Return(Variable('x')),
Stmt(Variable('x')),
]), None, 0))
expected_code = to_code([
LOAD_CONSTANT_FN, 0,
ASSIGN, 0,
LOAD_CONSTANT_FN, 0,
RETURN, 0])
expected_inner_bytecode = to_code([
LOAD_VAR, 0,
RETURN, 1,
LOAD_VAR, 0,
DISCARD_TOP, 0,
RETURN, 0,
])
assert bytecode.code == expected_code
assert bytecode.names == ['foo']
assert len(bytecode.constants_fn) == 1
inner = bytecode.constants_fn[0]
assert inner.code == expected_inner_bytecode
assert inner.names == ['x']
assert inner.constants_float == []
def test_stmt():
bytecode = compile_ast(Stmt(Variable('x')))
assert bytecode.code == to_code([LOAD_VAR, 0, DISCARD_TOP, 0, RETURN, 0])
assert bytecode.names == ['x']
assert bytecode.constants_float == []
def test_fn_def():
bytecode = compile_ast(FnDef('foo', [], Block([]), 'bar.js', 2),
co_name='foo', co_filename='bar.js', co_firstlineno=2)
expected_code = to_code([
LOAD_CONSTANT_FN, 0,
ASSIGN, 0,
LOAD_CONSTANT_FN, 0,
RETURN, 0])
expected_inner_bytecode = to_code([RETURN, 0])
assert bytecode.code == expected_code
assert bytecode.names == ['foo']
assert bytecode.co_name == 'foo'
assert bytecode.co_filename == 'bar.js'
assert bytecode.co_firstlineno == 2
assert len(bytecode.constants_fn) == 1
inner = bytecode.constants_fn[0]
assert inner.code == expected_inner_bytecode
assert inner.names == []
assert inner.constants_float == []
bytecode = compile_ast(FnDef('foo', ['x'], Block([
Stmt(BinOp('*', Variable('x'), Variable('x'))),
]), None, 0))
expected_code = to_code([
LOAD_CONSTANT_FN, 0,
ASSIGN, 0,
LOAD_CONSTANT_FN, 0,
RETURN, 0])
expected_inner_bytecode = to_code([
LOAD_VAR, 0,
LOAD_VAR, 0,
BINARY_MUL, 0,
DISCARD_TOP, 0,
RETURN, 0
])
assert bytecode.code == expected_code
assert bytecode.names == ['foo']
assert len(bytecode.constants_fn) == 1
inner = bytecode.constants_fn[0]
assert inner.code == expected_inner_bytecode
assert inner.names == ['x']
assert inner.constants_float == []
bytecode = compile_ast(FnDef('foo', ['y', 'x'], Block([
Stmt(BinOp('*', Variable('x'), Variable('x'))),
]), None, 0))
expected_code = to_code([
LOAD_CONSTANT_FN, 0,
ASSIGN, 0,
LOAD_CONSTANT_FN, 0,
RETURN, 0])
expected_inner_bytecode = to_code([
LOAD_VAR, 1,
LOAD_VAR, 1,
BINARY_MUL, 0,
DISCARD_TOP, 0,
RETURN, 0
])
assert bytecode.code == expected_code
assert bytecode.names == ['foo']
assert len(bytecode.constants_fn) == 1
inner = bytecode.constants_fn[0]
assert inner.names == ['y', 'x']
assert inner.code == expected_inner_bytecode
assert inner.constants_float == []
def test_const_num():
bytecode = compile_ast(ConstantNum(10.0))
assert bytecode.code == to_code([LOAD_CONSTANT_FLOAT, 0, RETURN, 0])
assert bytecode.names == []
assert bytecode.constants_float == [10.0]
def test_dis():
code = to_code([LOAD_CONSTANT_FLOAT, 1, RETURN, 0])
assert dis(code) == 'LOAD_CONSTANT_FLOAT 1\nRETURN 0'
