add test

Author: aquagull

test/legacy_test/test_cumsum_op.py

@@ -23,6 +23,7 @@
 
 import numpy as np
 from op_test import OpTest, convert_float_to_uint16
+from utils import dygraph_guard, static_guard
 
 import paddle
 import paddle.inference as paddle_infer
@@ -212,6 +213,12 @@ def set_attrs_input_output(self):
         self.out = self.x.cumsum(axis=0)
 
 
+class TestSumOpZeroSize(TestSumOp1):
+    def set_attrs_input_output(self):
+        self.x = np.random.random((1, 0, 4)).astype(self._dtype)
+        self.out = self.x.cumsum(axis=0)
+
+
 @unittest.skipIf(
     core.is_compiled_with_xpu(),
     "Skip XPU for complex dtype is not fully supported",
@@ -747,5 +754,40 @@ def test_fp16(self):
             paddle.disable_static()
 
 
+class TestSumOpDtypeAsPaddleDtype(unittest.TestCase):
+    def setUp(self):
+        self.shape = [2, 3, 4]
+        self.axis = 0
+        self.input_dtype = 'float32'
+        self.test_dtypes = [
+            paddle.int32,
+            paddle.int64,
+            paddle.float32,
+            paddle.float64,
+            paddle.bool,
+        ]
+
+    def test_dygraph(self):
+        with dygraph_guard():
+            x_paddle = paddle.ones(shape=self.shape, dtype=self.input_dtype)
+            for dtype_input in self.test_dtypes:
+                paddle_result = paddle.cumsum(
+                    x_paddle, axis=self.axis, dtype=dtype_input
+                )
+                self.assertEqual(paddle_result.dtype, dtype_input)
+
+    def test_static(self):
+        with static_guard():
+            for dtype_input in self.test_dtypes:
+                with paddle.static.program_guard(
+                    paddle.static.Program(), paddle.static.Program()
+                ):
+                    x = paddle.static.data(
+                        name='x', shape=self.shape, dtype=self.input_dtype
+                    )
+                    result = paddle.cumsum(x, axis=self.axis, dtype=dtype_input)
+                    self.assertEqual(result.dtype, dtype_input)
+
+
 if __name__ == '__main__':
     unittest.main()