experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #41390 from lgeiger:fix-pathlib-35-again

Browse Source 
PiperOrigin-RevId: 321236058
Change-Id: Ia802caa8772760541141871bada499b2e2132bc8
This commit is contained in:
TensorFlower Gardener 2020-07-14 14:39:07 -07:00
commit 4f3a8d84cf
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tensorflow/python/lib/io/file_io_test.py
View File

@ -39,6 +39,9 @@ class PathLike(object):
def __fspath__(self): def __fspath__(self):
return self.name return self.name
def __str__(self):
return self.name
run_all_path_types = parameterized.named_parameters( run_all_path_types = parameterized.named_parameters(
("str", os.path.join), ("str", os.path.join),
@ -197,7 +200,7 @@ class FileIoTest(test.TestCase, parameterized.TestCase):
dir_path = join(self._base_dir, "temp_dir/temp_dir1/temp_dir2") dir_path = join(self._base_dir, "temp_dir/temp_dir1/temp_dir2")
file_io.recursive_create_dir(dir_path) file_io.recursive_create_dir(dir_path)
file_io.recursive_create_dir(dir_path) # repeat creation file_io.recursive_create_dir(dir_path) # repeat creation
file_path = os.path.join(dir_path, "temp_file") file_path = os.path.join(str(dir_path), "temp_file")
file_io.FileIO(file_path, mode="w").write("testing") file_io.FileIO(file_path, mode="w").write("testing")
self.assertTrue(file_io.file_exists(file_path)) self.assertTrue(file_io.file_exists(file_path))
file_io.delete_recursively(os.path.join(self._base_dir, "temp_dir")) file_io.delete_recursively(os.path.join(self._base_dir, "temp_dir"))
@ -271,7 +274,7 @@ class FileIoTest(test.TestCase, parameterized.TestCase):
self.assertFalse(file_io.is_directory(dir_path)) self.assertFalse(file_io.is_directory(dir_path))
file_io.create_dir(dir_path) file_io.create_dir(dir_path)
self.assertTrue(file_io.is_directory(dir_path)) self.assertTrue(file_io.is_directory(dir_path))
file_path = join(dir_path, "test_file") file_path = join(str(dir_path), "test_file")
file_io.FileIO(file_path, mode="w").write("test") file_io.FileIO(file_path, mode="w").write("test")
# False for a file. # False for a file.
self.assertFalse(file_io.is_directory(file_path)) self.assertFalse(file_io.is_directory(file_path))
@ -285,11 +288,11 @@ class FileIoTest(test.TestCase, parameterized.TestCase):
file_io.create_dir(dir_path) file_io.create_dir(dir_path)
files = ["file1.txt", "file2.txt", "file3.txt"] files = ["file1.txt", "file2.txt", "file3.txt"]
for name in files: for name in files:
file_path = join(dir_path, name) file_path = join(str(dir_path), name)
file_io.FileIO(file_path, mode="w").write("testing") file_io.FileIO(file_path, mode="w").write("testing")
subdir_path = join(dir_path, "sub_dir") subdir_path = join(str(dir_path), "sub_dir")
file_io.create_dir(subdir_path) file_io.create_dir(subdir_path)
subdir_file_path = join(subdir_path, "file4.txt") subdir_file_path = join(str(subdir_path), "file4.txt")
file_io.FileIO(subdir_file_path, mode="w").write("testing") file_io.FileIO(subdir_file_path, mode="w").write("testing")
dir_list = file_io.list_directory(dir_path) dir_list = file_io.list_directory(dir_path)
self.assertItemsEqual(files + ["sub_dir"], dir_list) self.assertItemsEqual(files + ["sub_dir"], dir_list)
@ -319,7 +322,7 @@ class FileIoTest(test.TestCase, parameterized.TestCase):
def testWalkInOrder(self, join): def testWalkInOrder(self, join):
dir_path_str = os.path.join(self._base_dir, "test_dir") dir_path_str = os.path.join(self._base_dir, "test_dir")
dir_path = join(self._base_dir, "test_dir") dir_path = join(self._base_dir, "test_dir")
self._setupWalkDirectories(dir_path) self._setupWalkDirectories(dir_path_str)
# Now test the walk (in_order = True) # Now test the walk (in_order = True)
all_dirs = [] all_dirs = []
all_subdirs = [] all_subdirs = []
@ -389,7 +392,7 @@ class FileIoTest(test.TestCase, parameterized.TestCase):
file_path = join(self._base_dir, "temp_file") file_path = join(self._base_dir, "temp_file")
file_io.FileIO(file_path, mode="w").write("testing") file_io.FileIO(file_path, mode="w").write("testing")
file_statistics = file_io.stat(file_path) file_statistics = file_io.stat(file_path)
os_statistics = os.stat(file_path) os_statistics = os.stat(str(file_path))
self.assertEqual(7, file_statistics.length) self.assertEqual(7, file_statistics.length)
self.assertEqual( self.assertEqual(
int(os_statistics.st_mtime), int(file_statistics.mtime_nsec / 1e9)) int(os_statistics.st_mtime), int(file_statistics.mtime_nsec / 1e9))