diff --git a/python/paddle/fluid/tests/unittests/test_rnn_op.py b/python/paddle/fluid/tests/unittests/test_rnn_op.py index 44530bfba3061..79e33166bb6f8 100644 --- a/python/paddle/fluid/tests/unittests/test_rnn_op.py +++ b/python/paddle/fluid/tests/unittests/test_rnn_op.py @@ -173,5 +173,30 @@ def set_attrs(self): self.num_layers = 2 +class TestRNNOp6(TestRNNOp): + def set_attrs(self): + self.num_layers = 2 + self.is_bidirec = True + + +class TestRNNOp7(TestRNNOp): + def set_attrs(self): + self.num_layers = 2 + self.is_bidirec = True + self.is_test = True + + +class TestRNNOp8(TestRNNOp): + def set_attrs(self): + self.num_layers = 2 + self.is_bidirec = True + self.sequence_length = None + + +class TestRNNOp9(TestRNNOp): + def set_attrs(self): + self.num_layers = 3 + + if __name__ == '__main__': unittest.main()