about summary refs log tree commit diff stats
path: root/test/arch/mips32/unit/asm_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/arch/mips32/unit/asm_test.py')
-rw-r--r--test/arch/mips32/unit/asm_test.py20
1 files changed, 11 insertions, 9 deletions
diff --git a/test/arch/mips32/unit/asm_test.py b/test/arch/mips32/unit/asm_test.py
index da792874..2dcaf6fc 100644
--- a/test/arch/mips32/unit/asm_test.py
+++ b/test/arch/mips32/unit/asm_test.py
@@ -1,13 +1,15 @@
 import sys
 import os
 
-from miasm2.arch.mips32.arch import mn_mips32
-from miasm2.core import parse_asm
-from miasm2.expression.expression import *
-from miasm2.core import asmblock
-from elfesteem.strpatchwork import StrPatchwork
-from miasm2.analysis.machine import Machine
-from miasm2.jitter.csts import *
+from future.utils import viewitems
+
+from miasm.arch.mips32.arch import mn_mips32
+from miasm.core import parse_asm
+from miasm.expression.expression import *
+from miasm.core import asmblock
+from miasm.loader.strpatchwork import StrPatchwork
+from miasm.analysis.machine import Machine
+from miasm.jitter.csts import *
 
 
 reg_and_id = dict(mn_mips32.regs.all_regs_ids_byname)
@@ -30,10 +32,10 @@ class Asm_Test(object):
         loc_db.set_location_offset(loc_db.get_name_location("main"), 0x0)
         s = StrPatchwork()
         patches = asmblock.asm_resolve_final(mn_mips32, blocks, loc_db)
-        for offset, raw in patches.items():
+        for offset, raw in viewitems(patches):
             s[offset] = raw
 
-        s = str(s)
+        s = bytes(s)
         self.assembly = s
 
     def run(self):