add value compare

PaddlePaddle · Nov 25, 2022 · 7d87a36 · 7d87a36
1 parent 3e82eaa
commit 7d87a36
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/backends/mlu/tests/unittests/test_zero_dim_tensor_mlu.py b/backends/mlu/tests/unittests/test_zero_dim_tensor_mlu.py
@@ -434,6 +434,7 @@ def test_searchsorted(self):
         out = paddle.searchsorted(x, y)
 
         self.assertEqual(out.shape, [])
+        self.assertEqual(out.numpy(), 0)
 
 
 # Use to test API whose zero-dim input tensors don't have grad and not need to test backward in OpTest.

diff --git a/backends/npu/tests/unittests/test_zero_dim_tensor_npu.py b/backends/npu/tests/unittests/test_zero_dim_tensor_npu.py
@@ -434,6 +434,7 @@ def test_searchsorted(self):
         out = paddle.searchsorted(x, y)
 
         self.assertEqual(out.shape, [])
+        self.assertEqual(out.numpy(), 0)
 
 # Use to test API whose zero-dim input tensors don't have grad and not need to test backward in OpTest.
 class TestNoBackwardAPI(unittest.TestCase):