def test_find_ids(self): def f(): i = 0 x = 0 z = x + 3 # ID: myline return z # code = disassembler.dis(f) ids = find_ids(code) assert len(ids) == 1 myline = ids['myline'] opcodes_names = [opcode.__class__.__name__ for opcode in myline] assert opcodes_names == ['LOAD_FAST', 'LOAD_CONST', 'BINARY_ADD', 'STORE_FAST']