Skip to content

Commit da9b756

Browse files
committed
Added test for offset of array repr dtype string
1 parent a58249a commit da9b756

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

dpctl/tests/test_usm_ndarray_print.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,16 @@ def test_print_repr(self):
211211
x = dpt.arange(4, dtype="i4", sycl_queue=q)
212212
assert repr(x) == "usm_ndarray([0, 1, 2, 3], dtype=int32)"
213213

214+
dpt.set_print_options(linewidth=1)
215+
np.testing.assert_equal(
216+
repr(x),
217+
"usm_ndarray([0,"
218+
"\n 1,"
219+
"\n 2,"
220+
"\n 3],"
221+
"\n dtype=int32)",
222+
)
223+
214224
def test_print_repr_abbreviated(self):
215225
q = get_queue_or_skip()
216226

@@ -237,6 +247,19 @@ def test_print_repr_abbreviated(self):
237247
"\n [6, ..., 8]], dtype=int32)",
238248
)
239249

250+
dpt.set_print_options(linewidth=1)
251+
np.testing.assert_equal(
252+
repr(y),
253+
"usm_ndarray([[0,"
254+
"\n ...,"
255+
"\n 2],"
256+
"\n ...,"
257+
"\n [6,"
258+
"\n ...,"
259+
"\n 8]],"
260+
"\n dtype=int32)",
261+
)
262+
240263
@pytest.mark.parametrize(
241264
"dtype",
242265
[

0 commit comments

Comments
 (0)