Skip to content

Commit 460d5e5

Browse files
committed
Add arg validation tests for matrix_transpose
1 parent 1c4fede commit 460d5e5

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
# Data Parallel Control (dpctl)
2+
#
3+
# Copyright 2020-2023 Intel Corporation
4+
#
5+
# Licensed under the Apache License, Version 2.0 (the "License");
6+
# you may not use this file except in compliance with the License.
7+
# You may obtain a copy of the License at
8+
#
9+
# http://www.apache.org/licenses/LICENSE-2.0
10+
#
11+
# Unless required by applicable law or agreed to in writing, software
12+
# distributed under the License is distributed on an "AS IS" BASIS,
13+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14+
# See the License for the specific language governing permissions and
15+
# limitations under the License.
16+
17+
import pytest
18+
19+
import dpctl.tensor as dpt
20+
from dpctl.tests.helper import get_queue_or_skip
21+
22+
23+
def test_matrix_transpose_arg_validation():
24+
get_queue_or_skip()
25+
26+
X = dpt.ones(5, dtype="i4")
27+
28+
with pytest.raises(ValueError):
29+
dpt.matrix_transpose(X)
30+
31+
X = dict()
32+
with pytest.raises(TypeError):
33+
dpt.matrix_transpose(X)

0 commit comments

Comments
 (0)