diff --git a/framework/api/paddlebase/test_ones.py b/framework/api/paddlebase/test_ones.py index 9d4887f293..a4a3120e07 100644 --- a/framework/api/paddlebase/test_ones.py +++ b/framework/api/paddlebase/test_ones.py @@ -391,14 +391,15 @@ def test_ones32(): obj.run(res=res, shape=shape, dtype=dtype) -@pytest.mark.api_base_ones_exception +@pytest.mark.api_base_zeros_parameters def test_ones33(): """ shape_type=list,dtype='BOOL' """ shape = [1, 2, 3, 4] dtype = "BOOL" - obj.exception(mode="python", etype=TypeError, shape=shape, dtype=dtype) + res = np.ones(shape) + obj.run(res=res, shape=shape, dtype=dtype) @pytest.mark.api_base_ones_exception diff --git a/framework/api/paddlebase/test_zeros.py b/framework/api/paddlebase/test_zeros.py index b3df7511a1..5aa282dfa1 100644 --- a/framework/api/paddlebase/test_zeros.py +++ b/framework/api/paddlebase/test_zeros.py @@ -391,14 +391,15 @@ def test_zeros32(): obj.run(res=res, shape=shape, dtype=dtype) -@pytest.mark.api_base_zeros_exception +@pytest.mark.api_base_zeros_parameters def test_zeros33(): """ shape_type=list,dtype='BOOL' """ shape = [1, 2, 3, 4] dtype = "BOOL" - obj.exception(mode="python", etype=TypeError, shape=shape, dtype=dtype) + res = np.zeros(shape) + obj.run(res=res, shape=shape, dtype=dtype) @pytest.mark.api_base_zeros_exception