Fix the pytest error for async io

python/kvikio/kvikio/_lib/file_handle.pyx

@@ -86,6 +86,8 @@ cdef extern from "<kvikio/file_handle.hpp>" namespace "kvikio" nogil:
             size_t devPtr_offset,
             CUstream stream
         ) except +
+        bool is_compat_mode_preferred() except +
+        bool is_compat_mode_preferred_for_async() except +
 
 
 cdef class CuFile:
@@ -175,3 +177,9 @@ cdef class CuFile:
             dev_offset,
             stream,
         ))
+
+    def is_compat_mode_preferred(self) -> bool:
+        return self._handle.is_compat_mode_preferred()
+
+    def is_compat_mode_preferred_for_async(self) -> bool:
+        return self._handle.is_compat_mode_preferred_for_async()

python/kvikio/kvikio/cufile.py

@@ -430,3 +430,9 @@ def raw_write(
         to be a multiple of 4096 bytes. When GDS isn't used, this is less critical.
         """
         return self._handle.write(buf, size, file_offset, dev_offset)
+
+    def is_compat_mode_preferred(self) -> bool:
+        return self._handle.is_compat_mode_preferred()
+
+    def is_compat_mode_preferred_for_async(self) -> bool:
+        return self._handle.is_compat_mode_preferred_for_async()

python/kvikio/tests/test_async_io.py

@@ -29,8 +29,13 @@ def test_read_write(tmp_path, size):
     assert check_bit_flags(f.open_flags(), os.O_WRONLY)
     assert f.raw_write_async(a, stream.ptr).check_bytes_done() == a.nbytes
 
+    if f.is_compat_mode_preferred_for_async():
+        expected_except_msg = "Operation not permitted"
+    else:
+        expected_except_msg = "unsupported file open flags"
+
     # Try to read file opened in write-only mode
-    with pytest.raises(RuntimeError, match="Operation not permitted"):
+    with pytest.raises(RuntimeError, match=expected_except_msg):
         # The exception is raised when we call the raw_read_async API.
         future_stream = f.raw_read_async(a, stream.ptr)
         future_stream.check_bytes_done()