From e474bc17791693d5cee228b5bf35b8ef4cb48b43 Mon Sep 17 00:00:00 2001 From: wangchaochaohu Date: Mon, 17 Aug 2020 02:01:39 +0000 Subject: [PATCH] refine the code test=develop --- python/paddle/fluid/tests/unittests/test_chunk_op.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_chunk_op.py b/python/paddle/fluid/tests/unittests/test_chunk_op.py index 0c43baef0d7d5..8f9e899300a24 100644 --- a/python/paddle/fluid/tests/unittests/test_chunk_op.py +++ b/python/paddle/fluid/tests/unittests/test_chunk_op.py @@ -22,7 +22,7 @@ import paddle -class TestCunkOpError(unittest.TestCase): +class TestChunkOpError(unittest.TestCase): def test_errors(self): with program_guard(Program(), Program()): # The type of axis in chunk_op should be int or Variable. @@ -54,7 +54,7 @@ def test_axis_type_tensor(): self.assertRaises(TypeError, test_axis_type_tensor) -class API_TestSplit(unittest.TestCase): +class API_TestChunk(unittest.TestCase): def test_out(self): with fluid.program_guard(fluid.Program(), fluid.Program()): data1 = paddle.data('data1', shape=[4, 6, 6], dtype='float64') @@ -73,7 +73,7 @@ def test_out(self): self.assertTrue(np.allclose(ex_x2, r2)) -class API_TestSplit2(unittest.TestCase): +class API_TestChunk1(unittest.TestCase): def test_out(self): with fluid.program_guard(fluid.Program(), fluid.Program()): data1 = paddle.data('data1', shape=[4, 6, 6], dtype='float64') @@ -89,7 +89,7 @@ def test_out(self): self.assertTrue(np.allclose(ex_x2, r2)) -class API_TestDygraphSplit(unittest.TestCase): +class API_TestDygraphChunk(unittest.TestCase): def test_out1(self): with fluid.dygraph.guard(): input_1 = np.random.random([4, 6, 6]).astype("int32")