def test_bytecode(self):
source = """
def func(test):
if test == 1:
return 1
elif test == 2:
return 2
return 3
"""
code = disassemble(source, function=True)
# without line numbers
expected = """
LOAD_FAST 'test'
LOAD_CONST 1
COMPARE_OP <Compare.EQ: 2>
POP_JUMP_IF_FALSE <label_instr6>
LOAD_CONST 1
RETURN_VALUE
label_instr6:
LOAD_FAST 'test'
LOAD_CONST 2
COMPARE_OP <Compare.EQ: 2>
POP_JUMP_IF_FALSE <label_instr13>
LOAD_CONST 2
RETURN_VALUE
label_instr13:
LOAD_CONST 3
RETURN_VALUE
"""[1:].rstrip(" ")
self.check_dump_bytecode(code, expected)
# with line numbers
expected = """
L. 2 0: LOAD_FAST 'test'
1: LOAD_CONST 1
2: COMPARE_OP <Compare.EQ: 2>
3: POP_JUMP_IF_FALSE <label_instr6>
L. 3 4: LOAD_CONST 1
5: RETURN_VALUE
label_instr6:
L. 4 7: LOAD_FAST 'test'
8: LOAD_CONST 2
9: COMPARE_OP <Compare.EQ: 2>
10: POP_JUMP_IF_FALSE <label_instr13>
L. 5 11: LOAD_CONST 2
12: RETURN_VALUE
label_instr13:
L. 6 14: LOAD_CONST 3
15: RETURN_VALUE
"""[1:].rstrip(" ")
self.check_dump_bytecode(code, expected, lineno=True)
def test_bytecode(self):
source = """
def func(test):
if test == 1:
return 1
elif test == 2:
return 2
return 3
"""
code = disassemble(source, function=True)
# without line numbers
expected = """
LOAD_FAST 'test'
LOAD_CONST 1
COMPARE_OP <Compare.EQ: 2>
POP_JUMP_IF_FALSE <label_instr6>
LOAD_CONST 1
RETURN_VALUE
label_instr6:
LOAD_FAST 'test'
LOAD_CONST 2
COMPARE_OP <Compare.EQ: 2>
POP_JUMP_IF_FALSE <label_instr13>
LOAD_CONST 2
RETURN_VALUE
label_instr13:
LOAD_CONST 3
RETURN_VALUE
"""[1:].rstrip(" ")
self.check_dump_bytecode(code, expected)
# with line numbers
expected = """
L. 2 0: LOAD_FAST 'test'
1: LOAD_CONST 1
2: COMPARE_OP <Compare.EQ: 2>
3: POP_JUMP_IF_FALSE <label_instr6>
L. 3 4: LOAD_CONST 1
5: RETURN_VALUE
label_instr6:
L. 4 7: LOAD_FAST 'test'
8: LOAD_CONST 2
9: COMPARE_OP <Compare.EQ: 2>
10: POP_JUMP_IF_FALSE <label_instr13>
L. 5 11: LOAD_CONST 2
12: RETURN_VALUE
label_instr13:
L. 6 14: LOAD_CONST 3
15: RETURN_VALUE
"""[1:].rstrip(" ")
self.check_dump_bytecode(code, expected, lineno=True)
def test_concrete_bytecode(self):
source = """
def func(test):
if test == 1:
return 1
elif test == 2:
return 2
return 3
"""
code = disassemble(source, function=True)
code = code.to_concrete_bytecode()
# without line numbers
expected = f"""
0 LOAD_FAST 0
2 LOAD_CONST 1
4 COMPARE_OP 2
6 POP_JUMP_IF_FALSE {6 if OFFSET_AS_INSTRUCTION else 12}
8 LOAD_CONST 1
10 RETURN_VALUE
12 LOAD_FAST 0
14 LOAD_CONST 2
16 COMPARE_OP 2
18 POP_JUMP_IF_FALSE {12 if OFFSET_AS_INSTRUCTION else 24}
20 LOAD_CONST 2
22 RETURN_VALUE
24 LOAD_CONST 3
26 RETURN_VALUE
""".lstrip(
"\n"
)
self.check_dump_bytecode(code, expected)
# with line numbers
expected = f"""
L. 2 0: LOAD_FAST 0
2: LOAD_CONST 1
4: COMPARE_OP 2
6: POP_JUMP_IF_FALSE {6 if OFFSET_AS_INSTRUCTION else 12}
L. 3 8: LOAD_CONST 1
10: RETURN_VALUE
L. 4 12: LOAD_FAST 0
14: LOAD_CONST 2
16: COMPARE_OP 2
18: POP_JUMP_IF_FALSE {12 if OFFSET_AS_INSTRUCTION else 24}
L. 5 20: LOAD_CONST 2
22: RETURN_VALUE
L. 6 24: LOAD_CONST 3
26: RETURN_VALUE
""".lstrip(
"\n"
)
self.check_dump_bytecode(code, expected, lineno=True)
def test_concrete_bytecode(self):
source = """
def func(test):
if test == 1:
return 1
elif test == 2:
return 2
return 3
"""
code = disassemble(source, function=True)
code = code.to_concrete_bytecode()
# without line numbers
expected = """
0 LOAD_FAST 0
3 LOAD_CONST 1
6 COMPARE_OP 2
9 POP_JUMP_IF_FALSE 16
12 LOAD_CONST 1
15 RETURN_VALUE
16 LOAD_FAST 0
19 LOAD_CONST 2
22 COMPARE_OP 2
25 POP_JUMP_IF_FALSE 32
28 LOAD_CONST 2
31 RETURN_VALUE
32 LOAD_CONST 3
35 RETURN_VALUE
""".lstrip("\n")
self.check_dump_bytecode(code, expected)
# with line numbers
expected = """
L. 2 0: LOAD_FAST 0
3: LOAD_CONST 1
6: COMPARE_OP 2
9: POP_JUMP_IF_FALSE 16
L. 3 12: LOAD_CONST 1
15: RETURN_VALUE
L. 4 16: LOAD_FAST 0
19: LOAD_CONST 2
22: COMPARE_OP 2
25: POP_JUMP_IF_FALSE 32
L. 5 28: LOAD_CONST 2
31: RETURN_VALUE
L. 6 32: LOAD_CONST 3
35: RETURN_VALUE
""".lstrip("\n")
self.check_dump_bytecode(code, expected, lineno=True)
def test_bytecode_blocks(self):
source = """
def func(test):
if test == 1:
return 1
elif test == 2:
return 2
return 3
"""
code = disassemble(source, function=True)
code = ControlFlowGraph.from_bytecode(code)
# without line numbers
expected = textwrap.dedent("""
block1:
LOAD_FAST 'test'
LOAD_CONST 1
COMPARE_OP <Compare.EQ: 2>
POP_JUMP_IF_FALSE <block3>
-> block2
block2:
LOAD_CONST 1
RETURN_VALUE
block3:
LOAD_FAST 'test'
LOAD_CONST 2
COMPARE_OP <Compare.EQ: 2>
POP_JUMP_IF_FALSE <block5>
-> block4
block4:
LOAD_CONST 2
RETURN_VALUE
block5:
LOAD_CONST 3
RETURN_VALUE
""").lstrip()
self.check_dump_bytecode(code, expected)
# with line numbers
expected = textwrap.dedent("""
block1:
L. 2 0: LOAD_FAST 'test'
1: LOAD_CONST 1
2: COMPARE_OP <Compare.EQ: 2>
3: POP_JUMP_IF_FALSE <block3>
-> block2
block2:
L. 3 0: LOAD_CONST 1
1: RETURN_VALUE
block3:
L. 4 0: LOAD_FAST 'test'
1: LOAD_CONST 2
2: COMPARE_OP <Compare.EQ: 2>
3: POP_JUMP_IF_FALSE <block5>
-> block4
block4:
L. 5 0: LOAD_CONST 2
1: RETURN_VALUE
block5:
L. 6 0: LOAD_CONST 3
1: RETURN_VALUE
""").lstrip()
self.check_dump_bytecode(code, expected, lineno=True)
def test_concrete_bytecode(self):
source = """
def func(test):
if test == 1:
return 1
elif test == 2:
return 2
return 3
"""
code = disassemble(source, function=True)
code = code.to_concrete_bytecode()
# without line numbers
if WORDCODE:
expected = """
0 LOAD_FAST 0
2 LOAD_CONST 1
4 COMPARE_OP 2
6 POP_JUMP_IF_FALSE 12
8 LOAD_CONST 1
10 RETURN_VALUE
12 LOAD_FAST 0
14 LOAD_CONST 2
16 COMPARE_OP 2
18 POP_JUMP_IF_FALSE 24
20 LOAD_CONST 2
22 RETURN_VALUE
24 LOAD_CONST 3
26 RETURN_VALUE
""".lstrip("\n")
else:
expected = """
0 LOAD_FAST 0
3 LOAD_CONST 1
6 COMPARE_OP 2
9 POP_JUMP_IF_FALSE 16
12 LOAD_CONST 1
15 RETURN_VALUE
16 LOAD_FAST 0
19 LOAD_CONST 2
22 COMPARE_OP 2
25 POP_JUMP_IF_FALSE 32
28 LOAD_CONST 2
31 RETURN_VALUE
32 LOAD_CONST 3
35 RETURN_VALUE
""".lstrip("\n")
self.check_dump_bytecode(code, expected)
# with line numbers
if WORDCODE:
expected = """
L. 2 0: LOAD_FAST 0
2: LOAD_CONST 1
4: COMPARE_OP 2
6: POP_JUMP_IF_FALSE 12
L. 3 8: LOAD_CONST 1
10: RETURN_VALUE
L. 4 12: LOAD_FAST 0
14: LOAD_CONST 2
16: COMPARE_OP 2
18: POP_JUMP_IF_FALSE 24
L. 5 20: LOAD_CONST 2
22: RETURN_VALUE
L. 6 24: LOAD_CONST 3
26: RETURN_VALUE
""".lstrip("\n")
else:
expected = """
L. 2 0: LOAD_FAST 0
3: LOAD_CONST 1
6: COMPARE_OP 2
9: POP_JUMP_IF_FALSE 16
L. 3 12: LOAD_CONST 1
15: RETURN_VALUE
L. 4 16: LOAD_FAST 0
19: LOAD_CONST 2
22: COMPARE_OP 2
25: POP_JUMP_IF_FALSE 32
L. 5 28: LOAD_CONST 2
31: RETURN_VALUE
L. 6 32: LOAD_CONST 3
35: RETURN_VALUE
""".lstrip("\n")
self.check_dump_bytecode(code, expected, lineno=True)
def test_bytecode_blocks(self):
source = """
def func(test):
if test == 1:
return 1
elif test == 2:
return 2
return 3
"""
code = disassemble(source, function=True)
code = ControlFlowGraph.from_bytecode(code)
# without line numbers
expected = textwrap.dedent("""
block1:
LOAD_FAST 'test'
LOAD_CONST 1
COMPARE_OP <Compare.EQ: 2>
POP_JUMP_IF_FALSE <block3>
-> block2
block2:
LOAD_CONST 1
RETURN_VALUE
block3:
LOAD_FAST 'test'
LOAD_CONST 2
COMPARE_OP <Compare.EQ: 2>
POP_JUMP_IF_FALSE <block5>
-> block4
block4:
LOAD_CONST 2
RETURN_VALUE
block5:
LOAD_CONST 3
RETURN_VALUE
""").lstrip()
self.check_dump_bytecode(code, expected)
# with line numbers
expected = textwrap.dedent("""
block1:
L. 2 0: LOAD_FAST 'test'
1: LOAD_CONST 1
2: COMPARE_OP <Compare.EQ: 2>
3: POP_JUMP_IF_FALSE <block3>
-> block2
block2:
L. 3 0: LOAD_CONST 1
1: RETURN_VALUE
block3:
L. 4 0: LOAD_FAST 'test'
1: LOAD_CONST 2
2: COMPARE_OP <Compare.EQ: 2>
3: POP_JUMP_IF_FALSE <block5>
-> block4
block4:
L. 5 0: LOAD_CONST 2
1: RETURN_VALUE
block5:
L. 6 0: LOAD_CONST 3
1: RETURN_VALUE
""").lstrip()
self.check_dump_bytecode(code, expected, lineno=True)
