@@ -208,8 +208,8 @@ def test_print_repr(self):
208
208
"\n [6, 7, 8]])" ,
209
209
)
210
210
211
- x = dpt .arange (4 , dtype = "f2 " , sycl_queue = q )
212
- assert repr (x ) == "usm_ndarray([0. , 1. , 2. , 3. ], dtype=float16 )"
211
+ x = dpt .arange (4 , dtype = "i4 " , sycl_queue = q )
212
+ assert repr (x ) == "usm_ndarray([0, 1, 2, 3], dtype=int32 )"
213
213
214
214
def test_print_repr_abbreviated (self ):
215
215
q = get_queue_or_skip ()
@@ -218,8 +218,8 @@ def test_print_repr_abbreviated(self):
218
218
x = dpt .arange (9 , sycl_queue = q )
219
219
assert repr (x ) == "usm_ndarray([0, ..., 8])"
220
220
221
- y = dpt .asarray (x , dtype = "f2 " , copy = True )
222
- assert repr (y ) == "usm_ndarray([0. , ..., 8. ], dtype=float16 )"
221
+ y = dpt .asarray (x , dtype = "i4 " , copy = True )
222
+ assert repr (y ) == "usm_ndarray([0, ..., 8], dtype=int32 )"
223
223
224
224
x = dpt .reshape (x , (3 , 3 ))
225
225
np .testing .assert_equal (
@@ -232,9 +232,9 @@ def test_print_repr_abbreviated(self):
232
232
y = dpt .reshape (y , (3 , 3 ))
233
233
np .testing .assert_equal (
234
234
repr (y ),
235
- "usm_ndarray([[0. , ..., 2. ],"
235
+ "usm_ndarray([[0, ..., 2],"
236
236
"\n ...,"
237
- "\n [6. , ..., 8. ]], dtype=float16 )" ,
237
+ "\n [6, ..., 8]], dtype=int32 )" ,
238
238
)
239
239
240
240
@pytest .mark .parametrize (
0 commit comments