fix UT

Author: HydrogenSulfate

test/legacy_test/test_math_op_patch.py

@@ -684,7 +684,10 @@ def test_dygraph_vanilla_add(self):
  a_np = np.random.random((2, 3)).astype(np.float32) * 100
  b_f = lambda y: y + 1
  a = paddle.to_tensor(a_np)
- with self.assertRaises(TypeError):
+ with self.assertRaisesRegex(
+ TypeError,
+ r"__add__\(\): argument \(position 1\) must be int, float, bool or Tensor, but got ",
+ ):
  _ = a + b_f
 
  with self.assertRaisesRegex(
@@ -710,7 +713,10 @@ def test_dygraph_vanilla_mul(self):
  a_np = np.random.random((2, 3)).astype(np.float32) * 100
  b_f = lambda y: y + 1
  a = paddle.to_tensor(a_np)
- with self.assertRaises(TypeError):
+ with self.assertRaisesRegex(
+ TypeError,
+ r"__mul__\(\): argument \(position 1\) must be int, float, bool or Tensor, but got ",
+ ):
  _ = a * b_f
 
  with self.assertRaisesRegex(
@@ -732,7 +738,10 @@ def test_dygraph_vanilla_sub(self):
  a_np = np.random.random((2, 3)).astype(np.float32) * 100
  b_f = lambda y: y + 1
  a = paddle.to_tensor(a_np)
- with self.assertRaises(TypeError):
+ with self.assertRaisesRegex(
+ TypeError,
+ r"__sub__\(\): argument \(position 1\) must be int, float, bool or Tensor, but got ",
+ ):
  _ = a - b_f
 
  with self.assertRaisesRegex(
@@ -754,7 +763,10 @@ def test_dygraph_vanilla_pow(self):
  a_np = np.random.random((2, 3)).astype(np.float32) * 100
  b_f = lambda y: y + 1
  a = paddle.to_tensor(a_np)
- with self.assertRaises(TypeError):
+ with self.assertRaisesRegex(
+ TypeError,
+ r"__pow__\(\): argument \(position 1\) must be int, float, bool or Tensor, but got ",
+ ):
  _ = a**b_f
 
  with self.assertRaisesRegex(
@@ -776,7 +788,10 @@ def test_dygraph_vanilla_mod(self):
  a_np = np.random.random((2, 3)).astype(np.float32) * 100
  b_f = lambda y: y + 1
  a = paddle.to_tensor(a_np)
- with self.assertRaises(TypeError):
+ with self.assertRaisesRegex(
+ TypeError,
+ r"__mod__\(\): argument \(position 1\) must be int, float, bool or Tensor, but got ",
+ ):
  _ = a % b_f
 
  with self.assertRaisesRegex(
@@ -798,13 +813,16 @@ def test_dygraph_vanilla_matmul(self):
  a_np = np.random.random((2, 3)).astype(np.float32) * 100
  b_f = lambda y: y + 1
  a = paddle.to_tensor(a_np)
- with self.assertRaises(TypeError):
+ with self.assertRaisesRegex(
+ TypeError,
+ r"__matmul__\(\): argument \(position 1\) must be int, float, bool or Tensor, but got ",
+ ):
  _ = a @ b_f
 
  with self.assertRaisesRegex(
- ValueError, "Broadcast dimension mismatch"
+ ValueError, r"\(InvalidArgument\) Input\(Y\) has error dim"
  ):
- _ = a @ a.T
+ _ = a @ a
 
  def test_dygraph_custom_div(self):
  with dygraph_guard():
@@ -820,7 +838,10 @@ def test_dygraph_vanilla_div(self):
  a_np = np.random.random((2, 3)).astype(np.float32) * 100
  b_f = lambda y: y + 1
  a = paddle.to_tensor(a_np)
- with self.assertRaises(TypeError):
+ with self.assertRaisesRegex(
+ TypeError,
+ r"__div__\(\): argument \(position 1\) must be int, float, bool or Tensor, but got ",
+ ):
  _ = a / b_f
 
  with self.assertRaisesRegex(
@@ -842,7 +863,10 @@ def test_dygraph_vanilla_truediv(self):
  a_np = np.random.random((2, 3)).astype(np.float32) * 100
  b_f = lambda y: y + 1
  a = paddle.to_tensor(a_np)
- with self.assertRaises(TypeError):
+ with self.assertRaisesRegex(
+ TypeError,
+ r"__div__\(\): argument \(position 1\) must be int, float, bool or Tensor, but got ",
+ ):
  _ = a / b_f
 
  with self.assertRaisesRegex(
@@ -864,7 +888,10 @@ def test_dygraph_vanilla_floordiv(self):
  a_np = np.random.random((2, 3)).astype(np.float32) * 100
  b_f = lambda y: y + 1
  a = paddle.to_tensor(a_np)
- with self.assertRaises(TypeError):
+ with self.assertRaisesRegex(
+ TypeError,
+ r"__floordiv__\(\): argument \(position 1\) must be int, float, bool or Tensor, but got ",
+ ):
  _ = a // b_f
 
  with self.assertRaisesRegex(