545e8668b6c9de33638bb3257bb592eec53c377c
[oota-llvm.git] / bindings / python / llvm / tests / test_disassembler.py
1 from .base import TestBase
2
3 from ..disassembler import Disassembler
4
5 class TestDisassembler(TestBase):
6     def test_instantiate(self):
7          Disassembler('i686-apple-darwin9')
8
9     def test_basic(self):
10         sequence = '\x67\xe3\x81' # jcxz -127
11         triple = 'i686-apple-darwin9'
12
13         disassembler = Disassembler(triple)
14
15         count, s = disassembler.get_instruction(sequence)
16         self.assertEqual(count, 3)
17         self.assertEqual(s, '\tjcxz\t-127')
18
19     def test_get_instructions(self):
20         sequence = '\x67\xe3\x81\x01\xc7' # jcxz -127; addl %eax, %edi
21
22         disassembler = Disassembler('i686-apple-darwin9')
23
24         instructions = list(disassembler.get_instructions(sequence))
25         self.assertEqual(len(instructions), 2)
26
27         self.assertEqual(instructions[0], (0, 3, '\tjcxz\t-127'))
28         self.assertEqual(instructions[1], (3, 2, '\taddl\t%eax, %edi'))