[CodeStyle][task 10] enable ruff B017 rule in python/paddle/base (Pad…

…dlePaddle#58185)

pyproject.toml

@@ -103,14 +103,3 @@ ignore = [
 "test/dygraph_to_static/test_loop.py" = ["C416", "F821"]
 # Ignore unnecessary lambda in dy2st unittest test_lambda
 "test/dygraph_to_static/test_lambda.py" = ["PLC3002"]
-
-# B017
-"test/auto_parallel/spmd_rules/test_reshape_rule.py" = ["B017"]
-"test/dygraph_to_static/test_assert.py" = ["B017"]
-"test/legacy_test/test_cuda_max_memory_allocated.py" = ["B017"]
-"test/legacy_test/test_cuda_max_memory_reserved.py" = ["B017"]
-"test/legacy_test/test_cuda_memory_allocated.py" = ["B017"]
-"test/legacy_test/test_cuda_memory_reserved.py" = ["B017"]
-"test/legacy_test/test_eigvals_op.py" = ["B017"]
-"test/legacy_test/test_tensordot.py" = ["B017"]
-"test/legacy_test/test_top_k_v2_op.py" = ["B017"]

test/auto_parallel/spmd_rules/test_reshape_rule.py

@@ -246,7 +246,7 @@ def test_reshape_infer_forward(self):
         # shape: [6, 12, 48, 24] --> [3, 24, 6, -1, -1]
         # raise error
         self.attrs["shape"] = [3, 24, 6, -1, -1]
-        with self.assertRaises(BaseException):
+        with self.assertRaises(ValueError):
             self.rule.infer_forward(
                 self.x_dist_tensor_spec, self.attrs['shape']
             )

test/dygraph_to_static/test_assert.py

@@ -42,7 +42,7 @@ class TestAssertVariable(Dy2StTestBase):
     def _run(self, func, x, with_exception, to_static):
         paddle.jit.enable_to_static(to_static)
         if with_exception:
-            with self.assertRaises(BaseException):
+            with self.assertRaises(BaseException):  # noqa: B017
                 with base.dygraph.guard():
                     func(x)
         else:

test/legacy_test/test_cuda_max_memory_allocated.py

@@ -61,10 +61,10 @@ def test_max_memory_allocated_exception(self):
                 "gpu1",
             ]
             for device in wrong_device:
-                with self.assertRaises(BaseException):
+                with self.assertRaises(BaseException):  # noqa: B017
                     max_memory_allocated(device)
         else:
-            with self.assertRaises(BaseException):
+            with self.assertRaises(ValueError):
                 max_memory_allocated()
 
 

test/legacy_test/test_cuda_max_memory_reserved.py

@@ -61,10 +61,10 @@ def test_max_memory_reserved_exception(self):
                 "gpu1",
             ]
             for device in wrong_device:
-                with self.assertRaises(BaseException):
+                with self.assertRaises(BaseException):  # noqa: B017
                     max_memory_reserved(device)
         else:
-            with self.assertRaises(BaseException):
+            with self.assertRaises(ValueError):
                 max_memory_reserved()
 
 

test/legacy_test/test_cuda_memory_allocated.py

@@ -46,10 +46,10 @@ def test_memory_allocated_exception(self):
                 "gpu1",
             ]
             for device in wrong_device:
-                with self.assertRaises(BaseException):
+                with self.assertRaises(BaseException):  # noqa: B017
                     memory_allocated(device)
         else:
-            with self.assertRaises(BaseException):
+            with self.assertRaises(ValueError):
                 memory_allocated()
 
 

test/legacy_test/test_cuda_memory_reserved.py

@@ -46,10 +46,10 @@ def test_memory_reserved_exception(self):
                 "gpu1",
             ]
             for device in wrong_device:
-                with self.assertRaises(BaseException):
+                with self.assertRaises(BaseException):  # noqa: B017
                     memory_reserved(device)
         else:
-            with self.assertRaises(BaseException):
+            with self.assertRaises(ValueError):
                 memory_reserved()
 
 

test/legacy_test/test_eigvals_op.py

@@ -327,13 +327,13 @@ def test_cases(self):
     def test_error(self):
         paddle.disable_static()
         x = paddle.to_tensor([1])
-        with self.assertRaises(BaseException):
+        with self.assertRaises(ValueError):
             paddle.linalg.eigvals(x)
 
         self.input_dims = [1, 2, 3, 4]
         self.set_input_data()
         x = paddle.to_tensor(self.input_data)
-        with self.assertRaises(BaseException):
+        with self.assertRaises(ValueError):
             paddle.linalg.eigvals(x)
 
 

test/legacy_test/test_tensordot.py

@@ -342,9 +342,21 @@ def test_error(self):
         paddle.disable_static()
         x = paddle.to_tensor(self.x)
         y = paddle.to_tensor(self.y)
-        for axes in self.all_axes:
-            with self.assertRaises(BaseException):
-                paddle.tensordot(x, y, axes)
+
+        with self.assertRaises(TypeError):
+            paddle.tensordot(x, y, axes=self.all_axes[0])
+        with self.assertRaises(TypeError):
+            paddle.tensordot(x, y, axes=self.all_axes[1])
+        with self.assertRaises(AssertionError):
+            paddle.tensordot(x, y, axes=self.all_axes[2])
+        with self.assertRaises(IndexError):
+            paddle.tensordot(x, y, axes=self.all_axes[3])
+        with self.assertRaises(ValueError):
+            paddle.tensordot(x, y, axes=self.all_axes[4])
+        with self.assertRaises(AssertionError):
+            paddle.tensordot(x, y, axes=self.all_axes[5])
+        with self.assertRaises(AssertionError):
+            paddle.tensordot(x, y, axes=self.all_axes[6])
 
 
 class TestTensordotAPIAxesTypeFloat64(TestTensordotAPIAxesType):

test/legacy_test/test_top_k_v2_op.py

@@ -472,10 +472,10 @@ def test_cases(self):
     def test_errors(self):
         with paddle.base.dygraph.guard():
             x = paddle.to_tensor([1, 2, 3])
-            with self.assertRaises(BaseException):
+            with self.assertRaises(ValueError):
                 paddle.topk(x, k=-1)
 
-            with self.assertRaises(BaseException):
+            with self.assertRaises(ValueError):
                 paddle.topk(x, k=0)