def test_getfilesz(self): seg = DataElfSegment() self.assertRaises(InvalidArgument, seg.get_filesz) data = ByteArray("1234567890") data.memsz = 10 seg.set_data(data) self.assertEquals(seg.get_filesz(), 10) seg = HeaderElfSegment() self.assertRaises(Unprepared, seg.get_filesz) seg = SectionedElfSegment() self.assertEqual(seg.get_filesz(), 0)