Allow the list of previous checkpoints in the Saver to be set with a timestamp. Otherwise previous timestamp checkpoints could be kept forever.

Add a Stat function to the gfile module that returns the last modified time of a file.
Change: 113098506

1 files changed, 10 insertions, 0 deletions

diff --git a/tensorflow/python/platform/default/gfile_test.py b/tensorflow/python/platform/default/gfile_test.py
index 023dac56de..334db154d4 100644
--- a/tensorflow/python/platform/default/gfile_test.py
+++ b/tensorflow/python/platform/default/gfile_test.py
@@ -19,6 +19,7 @@ from __future__ import print_function
 
 import os
 import shutil
+import time
 
 from tensorflow.python.platform.default import _gfile as gfile
 from tensorflow.python.platform.default import _googletest as googletest
@@ -163,6 +164,15 @@ class FunctionTests(_BaseTest, googletest.TestCase):
     gfile.DeleteRecursively(self.tmp + "error_dir")
     self.assertFalse(gfile.Exists(self.tmp + "error_dir"))
 
+  def testStat(self):
+    with gfile.GFile(self.tmp + "test_stat", "w"):
+      pass
+    creation_time = time.time()
+    statinfo = gfile.Stat(self.tmp + "test_stat")
+    # Test the modification timestamp is within 20 seconds of closing the file.
+    self.assertLessEqual(statinfo.mtime, creation_time + 10)
+    self.assertGreaterEqual(statinfo.mtime, creation_time - 10)
+
 
 if __name__ == "__main__":
   googletest.main()