From ccb427cc9cecb99b262ca8b8291d366cd8b65e23 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andreas=20Kl=C3=B6ckner?= <inform@tiker.net>
Date: Fri, 15 Mar 2019 01:34:47 +0100
Subject: [PATCH] Adjust mempool tests for exposed leading bits
---
test/test_wrapper.py | 13 +++++++++----
1 file changed, 9 insertions(+), 4 deletions(-)
diff --git a/test/test_wrapper.py b/test/test_wrapper.py
index fa1e1d27..f7fa55ee 100644
--- a/test/test_wrapper.py
+++ b/test/test_wrapper.py
@@ -551,17 +551,22 @@ def test_mempool(ctx_factory):
pool.stop_holding()
-def test_mempool_2():
+def test_mempool_2(ctx_factory):
from pyopencl.tools import MemoryPool
from random import randrange
+ context = ctx_factory()
+ queue = cl.CommandQueue(context)
+
+ pool = MemoryPool(ImmediateAllocator(queue))
+
for i in range(2000):
s = randrange(1 << 31) >> randrange(32)
- bin_nr = MemoryPool.bin_number(s)
- asize = MemoryPool.alloc_size(bin_nr)
+ bin_nr = pool.bin_number(s)
+ asize = pool.alloc_size(bin_nr)
assert asize >= s, s
- assert MemoryPool.bin_number(asize) == bin_nr, s
+ assert pool.bin_number(asize) == bin_nr, s
assert asize < asize*(1+1/8)
--
GitLab