Skip to content

Commit b3ab3da

Browse files
committed
Added a test for matrix_transpose for coverage
1 parent 460d5e5 commit b3ab3da

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

dpctl/tests/test_usm_ndarray_linalg.py

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,14 +20,29 @@
2020
from dpctl.tests.helper import get_queue_or_skip
2121

2222

23-
def test_matrix_transpose_arg_validation():
23+
def test_matrix_transpose():
2424
get_queue_or_skip()
2525

26-
X = dpt.ones(5, dtype="i4")
26+
X = dpt.reshape(dpt.arange(2 * 3, dtype="i4"), (2, 3))
27+
res = dpt.matrix_transpose(X)
28+
expected_res = X.mT
29+
30+
assert expected_res.shape == res.shape
31+
assert expected_res.flags["C"] == res.flags["C"]
32+
assert expected_res.flags["F"] == res.flags["F"]
33+
assert dpt.all(X.mT == res)
2734

35+
36+
def test_matrix_transpose_arg_validation():
37+
get_queue_or_skip()
38+
39+
X = dpt.empty(5, dtype="i4")
2840
with pytest.raises(ValueError):
2941
dpt.matrix_transpose(X)
3042

3143
X = dict()
3244
with pytest.raises(TypeError):
3345
dpt.matrix_transpose(X)
46+
47+
X = dpt.empty((5, 5), dtype="i4")
48+
assert isinstance(dpt.matrix_transpose(X), dpt.usm_ndarray)

0 commit comments

Comments
 (0)