diff --git a/test/test_pytato_arraycontext.py b/test/test_pytato_arraycontext.py index d13b1ed90619c55d03b8f19d6bcdaa4adc723f65..fd3c6ea1aea8832b8d7799063bc2c51d9479e0e2 100644 --- a/test/test_pytato_arraycontext.py +++ b/test/test_pytato_arraycontext.py @@ -175,6 +175,11 @@ def test_pytato_actx_allocator(actx_factory, pass_allocator): return_value=False): actx = _PytatoPyOpenCLArrayContextForTests(base_actx.queue, allocator=alloc, use_memory_pool=use_memory_pool) + + from pyopencl.tools import ImmediateAllocator, MemoryPool + assert isinstance(actx.allocator, + MemoryPool if use_memory_pool else ImmediateAllocator) + f = actx.compile(twice) res = actx.to_numpy(f(99))