diff options
Diffstat (limited to 'tensorflow/python/lib/io')
-rw-r--r-- | tensorflow/python/lib/io/file_io.py | 4 | ||||
-rw-r--r-- | tensorflow/python/lib/io/file_io_test.py | 1 |
2 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/python/lib/io/file_io.py b/tensorflow/python/lib/io/file_io.py index ddd117e443..ace03e3d1b 100644 --- a/tensorflow/python/lib/io/file_io.py +++ b/tensorflow/python/lib/io/file_io.py @@ -146,9 +146,7 @@ class FileIO(object): def tell(self): """Returns the current position in the file.""" - if not self._read_check_passed: - raise errors.PermissionDeniedError(None, None, - "File isn't open for reading") + self._preread_check() return self._read_buf.Tell() def __enter__(self): diff --git a/tensorflow/python/lib/io/file_io_test.py b/tensorflow/python/lib/io/file_io_test.py index 0063eebb59..72931217d9 100644 --- a/tensorflow/python/lib/io/file_io_test.py +++ b/tensorflow/python/lib/io/file_io_test.py @@ -354,6 +354,7 @@ class FileIoTest(test.TestCase): file_path = os.path.join(self._base_dir, "temp_file") with file_io.FileIO(file_path, mode="r+") as f: f.write("testing1\ntesting2\ntesting3\n\ntesting5") + self.assertEqual(0, f.tell()) self.assertEqual("testing1\n", f.readline()) self.assertEqual(9, f.tell()) self.assertEqual("testing2\n", f.readline()) |