Skip to content

Commit fb579ad

Browse files
committed
Test Buffer.close with owner
Signed-off-by: Kamil Tokarski <[email protected]>
1 parent cf57d57 commit fb579ad

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

cuda_core/tests/test_memory.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -360,6 +360,17 @@ def test_memory_resource_and_owner_disallowed():
360360
Buffer.from_handle(ptr, 20, mr=DummyDeviceMemoryResource(Device()), owner=a)
361361

362362

363+
def test_owner_close():
364+
a = (ctypes.c_byte * 20)()
365+
ptr = ctypes.addressof(a)
366+
before = sys.getrefcount(a)
367+
buffer = Buffer.from_handle(ptr, 20, owner=a)
368+
assert sys.getrefcount(a) != before
369+
buffer.close()
370+
after = sys.getrefcount(a)
371+
assert after == before
372+
373+
363374
def test_buffer_dunder_dlpack():
364375
device = Device()
365376
device.set_current()

0 commit comments

Comments
 (0)