test_openpy.py
38 lines
| 1.4 KiB
| text/x-python
|
PythonLexer
Thomas Kluyver
|
r6452 | import io | |
| import os.path | |||
| import nose.tools as nt | |||
| from IPython.utils import openpy | |||
| mydir = os.path.dirname(__file__) | |||
| nonascii_path = os.path.join(mydir, '../../core/tests/nonascii.py') | |||
| def test_detect_encoding(): | |||
| f = open(nonascii_path, 'rb') | |||
| enc, lines = openpy.detect_encoding(f.readline) | |||
| nt.assert_equal(enc, 'iso-8859-5') | |||
| def test_read_file(): | |||
| read_specified_enc = io.open(nonascii_path, encoding='iso-8859-5').read() | |||
| read_detected_enc = openpy.read_py_file(nonascii_path, skip_encoding_cookie=False) | |||
| nt.assert_equal(read_detected_enc, read_specified_enc) | |||
Fernando Perez
|
r6563 | assert u'coding: iso-8859-5' in read_detected_enc | |
|
Thomas Kluyver
|
r6452 | ||
| read_strip_enc_cookie = openpy.read_py_file(nonascii_path, skip_encoding_cookie=True) | |||
|
Fernando Perez
|
r6563 | assert u'coding: iso-8859-5' not in read_strip_enc_cookie | |
|
Thomas Kluyver
|
r15516 | ||
| def test_source_to_unicode(): | |||
| with io.open(nonascii_path, 'rb') as f: | |||
| source_bytes = f.read() | |||
|
Thomas Kluyver
|
r16745 | nt.assert_equal(openpy.source_to_unicode(source_bytes, skip_encoding_cookie=False).splitlines(), | |
| source_bytes.decode('iso-8859-5').splitlines()) | |||
|
Thomas Kluyver
|
r15516 | ||
| source_no_cookie = openpy.source_to_unicode(source_bytes, skip_encoding_cookie=True) | |||
| nt.assert_not_in(u'coding: iso-8859-5', source_no_cookie) | |||
| def test_list_readline(): | |||
| l = ['a', 'b'] | |||
| readline = openpy._list_readline(l) | |||
| nt.assert_equal(readline(), 'a') | |||
| nt.assert_equal(readline(), 'b') | |||
| with nt.assert_raises(StopIteration): | |||
| readline() |