diff --git a/python/paddle/fluid/tests/unittests/test_chunk_op.py b/python/paddle/fluid/tests/unittests/test_chunk_op.py index 7124409727668f1c65dca437347a20e2f2db3157..f6bc1e4f03b19bbab930052f912580a2ca7ae73b 100644 --- a/python/paddle/fluid/tests/unittests/test_chunk_op.py +++ b/python/paddle/fluid/tests/unittests/test_chunk_op.py @@ -52,6 +52,14 @@ class TestChunkOpError(unittest.TestCase): self.assertRaises(TypeError, test_axis_type_tensor) + with paddle.fluid.dygraph.guard(): + + def test_0_chunks_tensor(): + x = paddle.uniform([1, 1, 1], dtype='float32') + paddle.chunk(x, chunks=0) + + self.assertRaises(ValueError, test_0_chunks_tensor) + class API_TestChunk(unittest.TestCase): def test_out(self):