diff --git a/pystencils_tests/test_buffer_gpu.py b/pystencils_tests/test_buffer_gpu.py index f9ee96e9b19c147d95f006de4221980900375cb1..f63254cd45797f2b2de8b6601fceb23bb0a94d54 100644 --- a/pystencils_tests/test_buffer_gpu.py +++ b/pystencils_tests/test_buffer_gpu.py @@ -39,7 +39,8 @@ def _generate_fields(dt=np.uint8, stencil_directions=1, layout='numpy'): gpu_src_arr = gpuarray.to_gpu(src_arr) gpu_dst_arr = gpuarray.zeros_like(gpu_src_arr) - gpu_buffer_arr = gpuarray.zeros(np.prod(src_arr.shape), dtype=dt) + size = int(np.prod(src_arr.shape)) + gpu_buffer_arr = gpuarray.zeros(size, dtype=dt) fields.append((src_arr, gpu_src_arr, gpu_dst_arr, gpu_buffer_arr)) return fields diff --git a/pytest.ini b/pytest.ini index ad6eb67d5e00832fd1c6860105b493a477834703..2795fb9d85e18838ddc963f307d232ef1a6365bf 100644 --- a/pytest.ini +++ b/pytest.ini @@ -39,7 +39,7 @@ exclude_lines = if __name__ == .__main__.: skip_covered = True -fail_under = 74 +fail_under = 75 [html] directory = coverage_report