test/test_arm_support.py

   1 #-------------------------------------------------------------------------------
   2 # elftools tests
   3 #
   4 # Eli Bendersky (eliben@gmail.com)
   5 # This code is in the public domain
   6 #-------------------------------------------------------------------------------
   7 try:
   8     import unittest2 as unittest
   9 except ImportError:
  10     import unittest
  11 import os
  12
  13 from utils import setup_syspath; setup_syspath()
  14 from elftools.elf.elffile import ELFFile
  15
  16 class TestARMSupport(unittest.TestCase):
  17     def test_hello(self):
  18         with open(os.path.join('test', 'testfiles',
  19                                'simple_gcc.elf.arm'), 'rb') as f:
  20             elf = ELFFile(f)
  21             self.assertEqual(elf.get_machine_arch(), 'ARM')
  22
  23             # Check some other properties of this ELF file derived from readelf
  24             self.assertEqual(elf['e_entry'], 0x8018)
  25             self.assertEqual(elf.num_sections(), 14)
  26             self.assertEqual(elf.num_segments(), 2)
  27
  28 if __name__ == '__main__':
  29     unittest.main()
  30