Fix compat issue

xia-mc · Dec 13, 2024 · 7c8f4ee · 7c8f4ee
1 parent 3ee9070
commit 7c8f4ee
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/tests/test_asm.py b/tests/test_asm.py
@@ -7,7 +7,7 @@
 from tests.benchmark import benchmark_ASM
 
 
-@unittest.skipUnless(platform.win32_edition() is not None, "ASM only support Windows now.")
+@unittest.skipUnless(platform.system() == "Windows", "ASM only support Windows now.")
 class TestASM(unittest.TestCase):
     @classmethod
     def setUpClass(cls):

diff --git a/tests/test_unsafe.py b/tests/test_unsafe.py
@@ -1,4 +1,5 @@
 import ctypes
+import platform
 import struct
 import sys
 import unittest
@@ -60,6 +61,7 @@ def test_set_and_get(self):
             self.assertEqual(result, 3)
             unsafe.free(ptr)
 
+    @unittest.skipUnless(platform.system() == "Windows", "ASM only support Windows now.")
     def test_callVoid(self):
         with ASM() as asm, Unsafe() as unsafe:
             voidFunc = asm.makeFunction(self.ks.asm("ret", as_bytes=True)[0])
@@ -68,6 +70,7 @@ def test_callVoid(self):
 
             asm.freeFunction(voidFunc)
 
+    @unittest.skipUnless(platform.system() == "Windows", "ASM only support Windows now.")
     def test_callInt(self):
         with ASM() as asm, Unsafe() as unsafe:
             func = asm.makeFunction(self.ks.asm("mov eax, 123456; ret", as_bytes=True)[0])
@@ -76,6 +79,7 @@ def test_callInt(self):
 
             asm.freeFunction(func)
 
+    @unittest.skipUnless(platform.system() == "Windows", "ASM only support Windows now.")
     def test_callLongLong(self):
         with ASM() as asm, Unsafe() as unsafe:
             func = asm.makeFunction(self.ks.asm("mov rax, 12345678910; ret", as_bytes=True)[0])
@@ -84,6 +88,7 @@ def test_callLongLong(self):
 
             asm.freeFunction(func)
 
+    @unittest.skipUnless(platform.system() == "Windows", "ASM only support Windows now.")
     def test_callCall(self):
         with ASM() as asm, Unsafe() as unsafe:
             func = asm.makeFunction(self.ks.asm("mov rax, 12345678910; ret", as_bytes=True)[0])
@@ -95,6 +100,7 @@ def test_callCall(self):
 
             asm.freeFunction(func)
 
+    @unittest.skipUnless(platform.system() == "Windows", "ASM only support Windows now.")
     def test_call_16bytes(self):
         with ASM() as asm, Unsafe() as unsafe:
             funcRet16Bytes = asm.makeFunction(self.ks.asm("""
@@ -116,6 +122,7 @@ def test_call_16bytes(self):
             asm.freeFunction(funcRet16Bytes)
             asm.freeFunction(funcProxy)
 
+    @unittest.skipUnless(platform.system() == "Windows", "ASM only support Windows now.")
     def test_call_size_zero(self):
         with ASM() as asm, Unsafe() as unsafe:
             voidFunc = asm.makeFunction(self.ks.asm("ret", as_bytes=True)[0])