Skip to content

Commit db420e9

Browse files
right context
1 parent 5941a9a commit db420e9

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/test_capi.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,10 @@ def test_array_get_shape(test_context, array_type):
199199
kernels=kernels,
200200
)
201201

202-
instance = array_type(np.array(range(3 * 5 * 7)).reshape((3, 5, 7)))
202+
instance = array_type(
203+
np.array(range(3 * 5 * 7)).reshape((3, 5, 7)),
204+
_context=test_context,
205+
)
203206

204207
expected_nd = 3
205208
result_nd = test_context.zeros((1,), dtype=np.int64)

0 commit comments

Comments
 (0)