Fix tests

Thriftpy · Apr 6, 2024 · 31f7a45 · 31f7a45
1 parent 1b624cd
commit 31f7a45
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/tests/test_cytransport.py b/tests/test_cytransport.py
@@ -2,14 +2,15 @@
 
 import pytest
 
-from thriftpy2._compat import PYPY
+from thriftpy2._compat import CYTHON
 
-pytestmark = pytest.mark.skipif(PYPY, reason="cython not enabled in pypy.")
 
-if not PYPY:
+if CYTHON:
     from thriftpy2.transport.framed import TCyFramedTransport
     from thriftpy2.transport.buffered import TCyBufferedTransport
     from thriftpy2.transport import TMemoryBuffer, TTransportException
+else:
+    pytest.skip("cython not enabled.", allow_module_level=True)
 
 
 def test_transport_mismatch():

diff --git a/tests/test_protocol_cybinary.py b/tests/test_protocol_cybinary.py
@@ -7,16 +7,16 @@
 
 import pytest
 
-from thriftpy2._compat import PYPY
+from thriftpy2._compat import CYTHON
 from thriftpy2.thrift import TDecodeException, TPayload, TType
 from thriftpy2.transport import TServerSocket, TSocket
 from thriftpy2.utils import hexlify
-if not PYPY:
+if CYTHON:
     from thriftpy2.protocol import cybin as proto
     from thriftpy2.transport.buffered import TCyBufferedTransport
     from thriftpy2.transport.memory import TCyMemoryBuffer
 else:
-    pytest.skip("cython not enabled in pypy.", allow_module_level=True)
+    pytest.skip("cython not enabled.", allow_module_level=True)
 
 
 class TItem(TPayload):