Skip to content

Commit 71c5b84

Browse files
Improve coverage of concat_axis_None
Incidentally, the added test also found a bug
1 parent dd28026 commit 71c5b84

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

dpctl/tests/test_usm_ndarray_manipulation.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -698,6 +698,14 @@ def test_concat_different_dtype():
698698
assert XY.shape == (5, 2)
699699
assert XY.sycl_queue == q
700700

701+
X1 = dpt.arange(10, dtype="i2", sycl_queue=q)
702+
Y1 = dpt.arange(5, dtype="i4", sycl_queue=q)
703+
704+
XY1 = dpt.concat([X1[::2], Y1[::-1]], axis=None)
705+
assert XY1.shape == (10,)
706+
assert XY1.sycl_queue == q
707+
assert XY1.dtype == Y1.dtype
708+
701709

702710
def test_concat_incorrect_ndim():
703711
q = get_queue_or_skip()

0 commit comments

Comments
 (0)