Regress python testcases must define expected value via unittest

regress/wrong_rip_arm.py

@@ -4,6 +4,8 @@ from unicorn import *
 from unicorn.x86_const import *
 from unicorn.arm_const import *
 
+import regress
+
 # adds  r1, #0x48
 # ldrsb r7, [r7, r7]
 # ldrsh r7, [r2, r1]
@@ -14,26 +16,23 @@ from unicorn.arm_const import *
 # strb  r7, [r5, #0xc]
 # ldr   r0, [pc, #0x1a0]
 binary1 = b'\x48\x31\xff\x57\x57\x5e\x5a\x48\xbf\x2f\x2f\x62\x69\x6e\x2f\x73\x68\x48\xc1\xef\x08\x57\x54\x5f\x6a\x3b\x58\x0f\x05'
-binary1 = b'\x48\x31\xff\x57'
+# binary1 = b'\x48\x31\xff\x57'
 #adds r1, #0x48
 #ldrsb  r7, [r7, r7]
 
-mu = Uc(UC_ARCH_ARM, UC_MODE_THUMB)
+class WrongRIPArm(regress.RegressTest):
 
-mu.mem_map(0, 2 * 1024 * 1024)
+    def runTest(self):
+        mu = Uc(UC_ARCH_ARM, UC_MODE_THUMB)
+        mu.mem_map(0, 2 * 1024 * 1024)
+        # write machine code to be emulated to memory
+        mu.mem_write(0, binary1)
+        mu.reg_write(UC_ARM_REG_R13, 1 * 1024 * 1024)
+        # emu for maximum 1 instruction.
+        mu.emu_start(0, len(binary1), 0, 1)
+        self.assertEqual(0x48, mu.reg_read(UC_ARM_REG_R1))
+        pos = mu.reg_read(UC_ARM_REG_R15)
+        self.assertEqual(0x2, pos)
 
-# write machine code to be emulated to memory
-mu.mem_write(0, binary1)
-
-mu.reg_write(ARM_REG_R13, 1*1024*1024)
-
-pos = 0
-
-# emu for maximum 1 instruction.
-mu.emu_start(pos, len(binary1), 0, 1)
-
-print("R1 = %x" % mu.reg_read(ARM_REG_R1))
-
-pos = mu.reg_read(ARM_REG_R15)
-
-print("RIP = %x" %pos)
+if __name__ == '__main__':
+    regress.main()