diff options
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/fileinput.py | 2 | ||||
-rw-r--r-- | Lib/test/test_fileinput.py | 10 |
2 files changed, 11 insertions, 1 deletions
diff --git a/Lib/fileinput.py b/Lib/fileinput.py index 27ccc3b..5c06627 100644 --- a/Lib/fileinput.py +++ b/Lib/fileinput.py @@ -184,7 +184,7 @@ class FileInput: """ def __init__(self, files=None, inplace=0, backup="", bufsize=0): - if type(files) == type(''): + if isinstance(files, basestring): files = (files,) else: if files is None: diff --git a/Lib/test/test_fileinput.py b/Lib/test/test_fileinput.py index 3a82c7c..285573c 100644 --- a/Lib/test/test_fileinput.py +++ b/Lib/test/test_fileinput.py @@ -157,3 +157,13 @@ try: verify(fi.lineno() == 6) finally: remove_tempfiles(t1, t2) + +if verbose: + print "15. Unicode filenames" +try: + t1 = writeTmp(1, ["A\nB"]) + fi = FileInput(files=unicode(t1, sys.getfilesystemencoding())) + lines = list(fi) + verify(lines == ["A\n", "B"]) +finally: + remove_tempfiles(t1) |