1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
import pytest
import opcode
def test_errors():
pytest.raises(ValueError, opcode.stack_effect, opcode.opmap["LOAD_CONST"])
pytest.raises(ValueError, opcode.stack_effect, opcode.opmap["BINARY_ADD"], 1)
pytest.raises(ValueError, opcode.stack_effect, 1231231)
def test_call_function():
assert opcode.stack_effect(opcode.opmap["CALL_FUNCTION"], 3) == -3
assert opcode.stack_effect(opcode.opmap["EXTENDED_ARG"], 3) == 0
def test_invalid_opcode():
pytest.raises(ValueError, opcode.stack_effect, 0)
pytest.raises(ValueError, opcode.stack_effect, 0, 0)
def test_jump():
assert opcode.stack_effect(opcode.opmap["FOR_ITER"], 0, jump=True) == -2
assert opcode.stack_effect(opcode.opmap["FOR_ITER"], 0, jump=False) == 1
assert opcode.stack_effect(opcode.opmap["FOR_ITER"], 0) == 1
|
