未验证 提交 f3ef748a 编写于 作者: A Ainavo 提交者: GitHub

[CodeStyle][UP005] remove deprecated unittest alias (#51834)

上级 e0eb5cf3
...@@ -36,6 +36,7 @@ select = [ ...@@ -36,6 +36,7 @@ select = [
"UP001", "UP001",
"UP003", "UP003",
"UP004", "UP004",
"UP005",
"UP007", "UP007",
"UP008", "UP008",
"UP010", "UP010",
......
...@@ -25,7 +25,7 @@ class TestIgnoreModule(unittest.TestCase): ...@@ -25,7 +25,7 @@ class TestIgnoreModule(unittest.TestCase):
def test_ignore_module(self): def test_ignore_module(self):
modules = [scipy, astor] modules = [scipy, astor]
ignore_module(modules) ignore_module(modules)
self.assertEquals( self.assertEqual(
[scipy, astor], [scipy, astor],
BUILTIN_LIKELY_MODULES[-2:], BUILTIN_LIKELY_MODULES[-2:],
'Failed to add modules that ignore transcription', 'Failed to add modules that ignore transcription',
......
...@@ -68,7 +68,7 @@ class TestGetGradOpDescPrimEnabled(unittest.TestCase): ...@@ -68,7 +68,7 @@ class TestGetGradOpDescPrimEnabled(unittest.TestCase):
self.fwd, self.no_grad_var, self.grad_sub_block self.fwd, self.no_grad_var, self.grad_sub_block
)[0] )[0]
) )
self.assertEquals(actual, self.desired_ops) self.assertEqual(actual, self.desired_ops)
if __name__ == '__main__': if __name__ == '__main__':
......
...@@ -80,7 +80,7 @@ class TestGetGradOpDescPrimEnabled(unittest.TestCase): ...@@ -80,7 +80,7 @@ class TestGetGradOpDescPrimEnabled(unittest.TestCase):
self.fwd, self.no_grad_var, self.grad_sub_block self.fwd, self.no_grad_var, self.grad_sub_block
)[0] )[0]
) )
self.assertEquals(actual, self.desired_ops) self.assertEqual(actual, self.desired_ops)
core._set_prim_backward_enabled(False) core._set_prim_backward_enabled(False)
......
...@@ -59,7 +59,7 @@ class TestGetGradOpDescPrimEnabled(unittest.TestCase): ...@@ -59,7 +59,7 @@ class TestGetGradOpDescPrimEnabled(unittest.TestCase):
self.fwd, self.no_grad_var, self.grad_sub_block self.fwd, self.no_grad_var, self.grad_sub_block
)[0] )[0]
) )
self.assertEquals(actual, self.desired_ops_no_skip) self.assertEqual(actual, self.desired_ops_no_skip)
core._set_prim_backward_enabled(False) core._set_prim_backward_enabled(False)
def test_get_grad_op_desc_with_skip(self): def test_get_grad_op_desc_with_skip(self):
...@@ -72,7 +72,7 @@ class TestGetGradOpDescPrimEnabled(unittest.TestCase): ...@@ -72,7 +72,7 @@ class TestGetGradOpDescPrimEnabled(unittest.TestCase):
)[0] )[0]
) )
core._remove_skip_comp_ops("tanh") core._remove_skip_comp_ops("tanh")
self.assertEquals(actual[0], self.desired_ops) self.assertEqual(actual[0], self.desired_ops)
core._set_prim_backward_enabled(False) core._set_prim_backward_enabled(False)
......
...@@ -100,7 +100,7 @@ class DotOpEmptyInput(unittest.TestCase): ...@@ -100,7 +100,7 @@ class DotOpEmptyInput(unittest.TestCase):
np_out = np.dot(data, data) np_out = np.dot(data, data)
pd_out = paddle.dot(x, y) pd_out = paddle.dot(x, y)
self.assertEquals(np_out, pd_out) self.assertEqual(np_out, pd_out)
def test_2d_input(self): def test_2d_input(self):
data = np.array([], dtype=np.float32) data = np.array([], dtype=np.float32)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册