Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix testcase for Randperm

This commit is contained in:
liyuxia 2023-01-30 15:03:46 +08:00
parent fdd9dcb992
commit de2aaa2381
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/st/ops/gpu/test_randperm.py
View File

@ -103,6 +103,6 @@ def test_randperm_uint64():
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_randperm_n_too_large():
with pytest.raises(RuntimeError) as info:
with pytest.raises(ValueError) as info:
randperm(1, 0, mindspore.int32, 2)
assert "n (2) cannot exceed max_length_ (1)" in str(info.value)
assert "For 'Randperm', input 'n' (2) cannot exceed 'max_length' (1)." in str(info.value)