diff options
Diffstat (limited to 'Lib/test/test_fileinput.py')
-rw-r--r-- | Lib/test/test_fileinput.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/Lib/test/test_fileinput.py b/Lib/test/test_fileinput.py index 565633f..5df810c 100644 --- a/Lib/test/test_fileinput.py +++ b/Lib/test/test_fileinput.py @@ -21,6 +21,7 @@ except ImportError: from io import BytesIO, StringIO from fileinput import FileInput, hook_encoded +from pathlib import Path from test.support import verbose, TESTFN, check_warnings from test.support import unlink as safe_unlink @@ -530,6 +531,20 @@ class FileInputTests(unittest.TestCase): self.assertRaises(StopIteration, next, fi) self.assertEqual(src.linesread, []) + def test_pathlib_file(self): + t1 = None + try: + t1 = Path(writeTmp(1, ["Pathlib file."])) + with FileInput(t1) as fi: + line = fi.readline() + self.assertEqual(line, 'Pathlib file.') + self.assertEqual(fi.lineno(), 1) + self.assertEqual(fi.filelineno(), 1) + self.assertEqual(fi.filename(), os.fspath(t1)) + finally: + remove_tempfiles(t1) + + class MockFileInput: """A class that mocks out fileinput.FileInput for use during unit tests""" |