def test_setdata(self): ef = DataElfSegment() data = ByteArray("foo") ef.set_data(data) self.assertEqual(ef.get_file_data(), data) ef = SectionedElfSegment(sections=[]) self.assertRaises(InvalidArgument, ef.set_data, data)