update UT

Author: HydrogenSulfate

test/legacy_test/test_empty.py

@@ -252,6 +252,17 @@ def test_Tensor_new_empty(self):
                 if isinstance(dtype, paddle.dtype):
                     self.assertEqual(x.dtype, dtype)
 
+                x = paddle.empty(
+                    [2],
+                ).new_empty(
+                    *shape,
+                    dtype=dtype,
+                    requires_grad=requires_grad,
+                    device=device,
+                    pin_memory=pin_memory,
+                )
+                self.assertEqual(x.shape, shape)
+
                 def new_empty(
                     x, shape, dtype, requires_grad, device, pin_memory
                 ):
@@ -283,6 +294,30 @@ def new_empty(
                 if isinstance(dtype, paddle.dtype):
                     self.assertEqual(x.dtype, dtype)
 
+                def new_empty_size_arg(
+                    x, shape, dtype, requires_grad, device, pin_memory
+                ):
+                    return x.new_empty(
+                        *shape,
+                        dtype=dtype,
+                        requires_grad=requires_grad,
+                        device=device,
+                        pin_memory=pin_memory,
+                    )
+
+                st_f = paddle.jit.to_static(
+                    new_empty_size_arg, full_graph=True, backend=None
+                )
+                x = st_f(
+                    paddle.randn([1]),
+                    shape,
+                    dtype=dtype,
+                    requires_grad=requires_grad,
+                    device=device,
+                    pin_memory=pin_memory,
+                )
+                self.assertEqual(x.shape, shape)
+
 
 class TestCreationOut(unittest.TestCase):
     def setUp(self):

test/legacy_test/test_math_op_patch_pir.py

@@ -788,6 +788,11 @@ def test_new_xxx(self):
                 (output_x,) = exe.run(main_program, fetch_list=[x_new])
                 self.assertEqual(output_x.shape, (2, 3))
 
+                x_new = x.new_zeros(2, 3)
+                self.assertEqual(x_new.shape, [2, 3])
+                (output_x,) = exe.run(main_program, fetch_list=[x_new])
+                self.assertEqual(output_x.shape, (2, 3))
+
         # test mT with dynamic shape
         with paddle.pir_utils.IrGuard():
             main_program, exe, program_guard = new_program()

test/legacy_test/test_ones.py

@@ -246,6 +246,17 @@ def test_Tensor_new_ones(self):
                 if isinstance(dtype, paddle.dtype):
                     self.assertEqual(x.dtype, dtype)
 
+                x = paddle.ones(
+                    [2],
+                ).new_ones(
+                    *shape,
+                    dtype=dtype,
+                    requires_grad=requires_grad,
+                    device=device,
+                    pin_memory=pin_memory,
+                )
+                self.assertEqual(x.shape, shape)
+
                 def new_ones(
                     x, shape, dtype, requires_grad, device, pin_memory
                 ):
@@ -277,6 +288,30 @@ def new_ones(
                 if isinstance(dtype, paddle.dtype):
                     self.assertEqual(x.dtype, dtype)
 
+                def new_ones_size_arg(
+                    x, shape, dtype, requires_grad, device, pin_memory
+                ):
+                    return x.new_ones(
+                        *shape,
+                        dtype=dtype,
+                        requires_grad=requires_grad,
+                        device=device,
+                        pin_memory=pin_memory,
+                    )
+
+                st_f = paddle.jit.to_static(
+                    new_ones_size_arg, full_graph=True, backend=None
+                )
+                x = st_f(
+                    paddle.randn([1]),
+                    shape,
+                    dtype=dtype,
+                    requires_grad=requires_grad,
+                    device=device,
+                    pin_memory=pin_memory,
+                )
+                self.assertEqual(x.shape, shape)
+
 
 class TestCreationOut(unittest.TestCase):
     def setUp(self):

test/legacy_test/test_zeros.py

@@ -246,6 +246,17 @@ def test_Tensor_new_zeros(self):
                 if isinstance(dtype, paddle.dtype):
                     self.assertEqual(x.dtype, dtype)
 
+                x = paddle.zeros(
+                    [2],
+                ).new_zeros(
+                    *shape,
+                    dtype=dtype,
+                    requires_grad=requires_grad,
+                    device=device,
+                    pin_memory=pin_memory,
+                )
+                self.assertEqual(x.shape, shape)
+
                 def new_zeros(
                     x, shape, dtype, requires_grad, device, pin_memory
                 ):
@@ -277,6 +288,30 @@ def new_zeros(
                 if isinstance(dtype, paddle.dtype):
                     self.assertEqual(x.dtype, dtype)
 
+                def new_zeros_size_arg(
+                    x, shape, dtype, requires_grad, device, pin_memory
+                ):
+                    return x.new_zeros(
+                        *shape,
+                        dtype=dtype,
+                        requires_grad=requires_grad,
+                        device=device,
+                        pin_memory=pin_memory,
+                    )
+
+                st_f = paddle.jit.to_static(
+                    new_zeros_size_arg, full_graph=True, backend=None
+                )
+                x = st_f(
+                    paddle.randn([1]),
+                    shape,
+                    dtype=dtype,
+                    requires_grad=requires_grad,
+                    device=device,
+                    pin_memory=pin_memory,
+                )
+                self.assertEqual(x.shape, shape)
+
 
 class TestCreationOut(unittest.TestCase):
     def setUp(self):