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) assert u'coding: iso-8859-5' in read_detected_enc read_strip_enc_cookie = openpy.read_py_file(nonascii_path, skip_encoding_cookie=True) assert u'coding: iso-8859-5' not in read_strip_enc_cookie def test_source_to_unicode(): with io.open(nonascii_path, 'rb') as f: source_bytes = f.read() nt.assert_equal(openpy.source_to_unicode(source_bytes, skip_encoding_cookie=False).splitlines(), source_bytes.decode('iso-8859-5').splitlines()) 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()