@@ -191,13 +191,13 @@ def test_print_str_abbreviated(self):
191
191
def test_print_repr (self ):
192
192
q = get_queue_or_skip ()
193
193
194
- x = dpt .asarray (0 , sycl_queue = q )
194
+ x = dpt .asarray (0 , dtype = "int64" , sycl_queue = q )
195
195
assert repr (x ) == "usm_ndarray(0)"
196
196
197
197
x = dpt .asarray ([np .nan , np .inf ], sycl_queue = q )
198
198
assert repr (x ) == "usm_ndarray([nan, inf])"
199
199
200
- x = dpt .arange (9 , sycl_queue = q )
200
+ x = dpt .arange (9 , sycl_queue = q , dtype = "int64" )
201
201
assert repr (x ) == "usm_ndarray([0, 1, 2, 3, 4, 5, 6, 7, 8])"
202
202
203
203
x = dpt .reshape (x , (3 , 3 ))
@@ -208,18 +208,18 @@ 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 ()
216
216
217
217
dpt .set_print_options (threshold = 0 , edgeitems = 1 )
218
- x = dpt .arange (9 , sycl_queue = q )
218
+ x = dpt .arange (9 , dtype = "int64" , 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