add some test

PaddlePaddle · Oct 15, 2023 · 7bc46e3 · 7bc46e3
1 parent 211db2b
commit 7bc46e3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/legacy_test/test_math_op_patch_pir.py b/test/legacy_test/test_math_op_patch_pir.py
@@ -21,6 +21,12 @@
 
 
 class TestMathOpPatchesPir(unittest.TestCase):
+    def test_item(self):
+        with paddle.pir_utils.IrGuard():
+            x = paddle.static.data(name='x', shape=[3, 2, 1])
+            with self.assertRaises(TypeError):
+                x.item()
+
     def test_some_dim(self):
         with paddle.pir_utils.IrGuard():
             x = paddle.static.data(name='x', shape=[3, 2, 1])