add more test for rnn

PaddlePaddle · Nov 10, 2021 · 0fa3088 · 0fa3088 · paddle-bot-old · Nov 10, 2021
1 parent 49f7bb2
commit 0fa3088
Showing 1 changed file with 25 additions and 0 deletions.
diff --git 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()