Merge pull request PaddlePaddle#4 from feifei-111/test_cache_program

update test_cache_program

test/dygraph_to_static/test_cache_program.py

@@ -16,6 +16,7 @@
 from collections import Counter
 
 import numpy as np
+from dygraph_to_static_util import dy2static_unittest
 from test_fetch_feed import Linear, Pool2D
 
 import paddle
@@ -24,6 +25,7 @@
 from paddle.jit.dy2static import convert_to_static
 
 
+@dy2static_unittest
 class TestCacheProgram(unittest.TestCase):
     def setUp(self):
         self.batch_num = 5
@@ -36,7 +38,7 @@ def test_cache(self):
         with fluid.dygraph.guard(fluid.CPUPlace()):
             static_net = self.dygraph_class()
             for batch_id in range(self.batch_num):
-                out = static_net(self.data)
+                out = static_net(paddle.to_tensor(self.data))
                 # Check outputs
                 prev_out = cur_out
                 cur_out = out

test/dygraph_to_static/test_fallback.py

@@ -16,6 +16,7 @@
 import unittest
 
 import numpy as np
+from dygraph_to_static_util import ast_only_test
 
 import paddle
 
@@ -84,6 +85,7 @@ def test_case_net_fallback(self):
             u_net(self.x).numpy(),
         )
 
+    @ast_only_test
     def test_case_net_error(self):
         s_net = SuppportNet()
         u_net = UnsuppportNet()
@@ -110,6 +112,7 @@ def test_case_training(self):
         np.testing.assert_allclose(u_net(self.x).numpy(), [1, 1])
         assert u_net.training is False, "Training must be false."
 
+    @ast_only_test
     def test_case_save_error(self):
         """
         test the save will raise error.