From: Eli Bendersky <eliben@gmail.com>
Date: Wed, 5 Sep 2018 12:35:33 +0000 (-0700)
Subject: Simplify arm reloc test file
X-Git-Tag: v0.26~38
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=e762568c490f26ed7f75afc6a48c339bd43b5689;p=pyelftools.git

Simplify arm reloc test file
---

diff --git a/test_arm_call_reloc.py b/test_arm_call_reloc.py
index cb52419..8a2eac5 100755
--- a/test_arm_call_reloc.py
+++ b/test_arm_call_reloc.py
@@ -10,6 +10,7 @@
 #-------------------------------------------------------------------------------
 import os
 import sys
+import unittest
 
 from elftools.common.py3compat import BytesIO
 from elftools.elf.elffile import ELFFile
@@ -25,11 +26,12 @@ def do_relocation(rel_elf):
 
     rel = rel_elf.get_section_by_name('.rel.text')
     rh.apply_section_relocations(stream, rel)
+    return data.getvalue()
 
-    stream.seek(0)
-    data = stream.readlines()
+    #stream.seek(0)
+    #data = stream.readlines()
 
-    return data
+    #return data
 
 
 class TestARMRElocation(unittest.TestCase):
@@ -41,5 +43,5 @@ class TestARMRElocation(unittest.TestCase):
             elf = ELFFile(f)
 
             # Comparison of '.text' section data
-            self.assertEquals(do_relocation(rel_elf).pop(),
+            self.assertEquals(do_relocation(rel_elf),
                               elf.get_section_by_name('.text').data())