Skip to content

Commit 5ef035e

Browse files
More tests
1 parent 726b359 commit 5ef035e

File tree

1 file changed

+21
-2
lines changed

1 file changed

+21
-2
lines changed

dpctl/tests/test_sycl_kernel_submit.py

+21-2
Original file line numberDiff line numberDiff line change
@@ -247,11 +247,30 @@ def test_submit_async():
247247
assert np.array_equal(Xnp[:, :n], Xref[:, :n])
248248

249249

250+
def _check_kernel_arg_type_instance(kati):
251+
assert isinstance(kati.name, str)
252+
assert isinstance(kati.value, int)
253+
assert isinstance(repr(kati), str)
254+
assert isinstance(str(kati), str)
255+
256+
250257
def test_kernel_arg_type():
251258
"""
252259
Check that enum values for kernel_arg_type start at 0,
253260
as numba_dpex expects. The next enumerated type must
254261
have next value.
255262
"""
256-
assert kernel_arg_type.dpctl_int8.value == 0
257-
assert kernel_arg_type.dpctl_uint8.value == 1
263+
assert isinstance(kernel_arg_type.__name__, str)
264+
assert isinstance(repr(kernel_arg_type), str)
265+
assert isinstance(str(kernel_arg_type), str)
266+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_int8)
267+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_uint8)
268+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_int16)
269+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_uint16)
270+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_int32)
271+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_uint32)
272+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_int64)
273+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_uint64)
274+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_float32)
275+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_float64)
276+
_check_kernel_arg_type_instance(kernel_arg_type.dpctl_void_ptr)

0 commit comments

Comments
 (0)