Skip to content

Commit 0a79c69

Browse files
committed
graphics: cover chained mapped buffer context
1 parent 591f0c7 commit 0a79c69

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

cuda_core/tests/test_graphics.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -318,6 +318,16 @@ def test_resource_context_manager_can_map_inside_scope(self):
318318
assert isinstance(buf, Buffer)
319319
assert buf.handle != 0
320320

321+
def test_chained_map_context_manager_unmaps(self):
322+
with _gl_context_and_buffer(nbytes=4096) as (gl_buf, _):
323+
stream = _create_stream()
324+
with GraphicsResource.from_gl_buffer(gl_buf, flags="write_discard").map(stream=stream) as buf:
325+
assert isinstance(buf, Buffer)
326+
assert buf.handle != 0
327+
assert buf.size > 0
328+
assert buf.handle == 0
329+
assert buf.size == 0
330+
321331
def test_map_with_stream(self):
322332
with _gl_context_and_buffer(nbytes=4096) as (gl_buf, nbytes):
323333
stream = _create_stream()

0 commit comments

Comments
 (0)