@@ -247,11 +247,30 @@ def test_submit_async():
247
247
assert np .array_equal (Xnp [:, :n ], Xref [:, :n ])
248
248
249
249
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
+
250
257
def test_kernel_arg_type ():
251
258
"""
252
259
Check that enum values for kernel_arg_type start at 0,
253
260
as numba_dpex expects. The next enumerated type must
254
261
have next value.
255
262
"""
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