bindings/python/llvm/tests/test_core.py

   1 from .base import TestBase
   2 from ..core import OpCode
   3 from ..core import MemoryBuffer
   4
   5 class TestCore(TestBase):
   6     def test_opcode(self):
   7         self.assertTrue(hasattr(OpCode, 'Ret'))
   8         self.assertTrue(isinstance(OpCode.Ret, OpCode))
   9         self.assertEqual(OpCode.Ret.value, 1)
  10
  11         op = OpCode.from_value(1)
  12         self.assertTrue(isinstance(op, OpCode))
  13         self.assertEqual(op, OpCode.Ret)
  14
  15     def test_memory_buffer_create_from_file(self):
  16         source = self.get_test_binary()
  17
  18         MemoryBuffer(filename=source)
  19
  20     def test_memory_buffer_failing(self):
  21         with self.assertRaises(Exception):
  22             MemoryBuffer(filename="/hopefully/this/path/doesnt/exist")
  23
  24     def test_memory_buffer_len(self):
  25         source = self.get_test_file()
  26         m = MemoryBuffer(filename=source)
  27         self.assertEqual(len(m), 50)