[PIR] skip bad case in test_cache_program.py and revert some changes (#…

…59697)

* fix

* fix

test/dygraph_to_static/test_cache_program.py

@@ -18,6 +18,9 @@
 import numpy as np
 from dygraph_to_static_utils import (
     Dy2StTestBase,
+    IrMode,
+    ToStaticMode,
+    disable_test_case,
     enable_to_static_guard,
     test_ast_only,
     test_legacy_and_pt_and_pir,
@@ -172,6 +175,7 @@ def sum_under_while(limit):
     return ret_sum
 
 
+@disable_test_case((ToStaticMode.AST, IrMode.PT))
 class TestToOutputWithCache(Dy2StTestBase):
     def test_output(self):
         ret = paddle.jit.to_static(sum_even_until_limit)(80, 10)

test/dygraph_to_static/test_write_python_container.py

@@ -14,10 +14,7 @@
 
 import unittest
 
-from dygraph_to_static_utils import (
-    Dy2StTestBase,
-    test_sot_only,
-)
+from dygraph_to_static_utils import Dy2StTestBase, test_ast_only, test_sot_only
 
 import paddle
 
@@ -123,6 +120,16 @@ def test_write_container_sot(self):
         out_dygraph = self.get_raw_value(self.func(input), self.getitem_path)
         self.assertEqual(out_static, out_dygraph)
 
+    @test_ast_only
+    def test_write_container(self):
+        func_static = paddle.jit.to_static(self.func)
+        input = paddle.to_tensor([1, 2, 3])
+        out_static = self.get_raw_value(
+            func_static(input), self.getitem_path
+        ).item()
+        out_dygraph = self.get_raw_value(self.func(input), self.getitem_path)
+        self.assertEqual(out_static, out_dygraph)
+
 
 class TestLoopWriteContainerList(TestWriteContainer):
     def set_func(self):