diff --git a/pyproject.toml b/pyproject.toml index 68887b9df0812edbb566614ce24cfd566666d343..15999a333509ab11e70a3deaa958c91c2a7ab46b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -100,7 +100,7 @@ select = [ "B012", "B013", "B014", - # "B015", + "B015", "B016", # "B017", "B018", diff --git a/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_assign.py b/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_assign.py index eb9117772b3fdb10830cd452e380d132d2332121..87064a45a49ffb678a8e4974b6f3a07700e9e031 100644 --- a/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_assign.py +++ b/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_assign.py @@ -65,8 +65,7 @@ class TestDistAssign(unittest.TestCase): for op in ops: if op.type == "assign": dist_op = dist_context.get_dist_op_for_program(op) - dist_op.dist_attr.impl_type == "assign" - dist_op.dist_attr.impl_idx == 0 + assert dist_op.dist_attr.impl_type == "assign" x_name = op.input_arg_names[0] out_name = op.output_arg_names[0] diff --git a/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_scale.py b/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_scale.py index 2bee2285b58ab2c84fe8c4e5a678a63d837b5f66..b68131e361ec04469bae594309b35b23bfde841a 100644 --- a/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_scale.py +++ b/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_scale.py @@ -60,8 +60,8 @@ class TestDistScale(unittest.TestCase): ops = dist_main_prog.global_block().ops scale_op = ops[0] dist_op = dist_context.get_dist_op_for_program(scale_op) - dist_op.dist_attr.impl_type == "scale" - dist_op.dist_attr.impl_idx == 0 + assert dist_op.dist_attr.impl_type == "scale" + assert dist_op.dist_attr.impl_idx == 0 in_name = scale_op.input_arg_names[0] out_name = scale_op.output_arg_names[0] diff --git a/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_shape.py b/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_shape.py index 5c4ecfcfabbc3d8d2de64b56f64d861453049e2a..0322a817934fdfd1132eab3489c996438b3fa270 100644 --- a/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_shape.py +++ b/python/paddle/fluid/tests/unittests/auto_parallel/test_dist_shape.py @@ -60,8 +60,8 @@ class TestDistShape(unittest.TestCase): ops = dist_main_prog.global_block().ops shape_op = ops[0] dist_op = dist_context.get_dist_op_for_program(shape_op) - dist_op.dist_attr.impl_type == "shape" - dist_op.dist_attr.impl_idx == 0 + assert dist_op.dist_attr.impl_type == "shape" + assert dist_op.dist_attr.impl_idx == 0 in_name = shape_op.input_arg_names[0] out_name = shape_op.output_arg_names[0] diff --git a/python/paddle/fluid/tests/unittests/test_clip_op.py b/python/paddle/fluid/tests/unittests/test_clip_op.py index 9b6ddcee80e7744e7e410671c597815c6525a588..979ce6e7cc991691ad5860914237490a47d22ace 100644 --- a/python/paddle/fluid/tests/unittests/test_clip_op.py +++ b/python/paddle/fluid/tests/unittests/test_clip_op.py @@ -113,7 +113,7 @@ class TestCase5(TestClipOp): class TestCase6(TestClipOp): def initTestCase(self): - self.dtype == np.float16 + self.dtype = np.float16 self.shape = (4, 8, 8) self.max = 0.7 self.min = 0.2