40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
|
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():
|
||
|
|
with open(nonascii_path, "rb") as f:
|
||
|
|
enc, lines = openpy.detect_encoding(f.readline)
|
||
|
|
nt.assert_equal(enc, "iso-8859-5")
|
||
|
|
|
||
|
|
|
||
|
|
def test_read_file():
|
||
|
|
with io.open(nonascii_path, encoding="iso-8859-5") as f:
|
||
|
|
read_specified_enc = f.read()
|
||
|
|
read_detected_enc = openpy.read_py_file(nonascii_path, skip_encoding_cookie=False)
|
||
|
|
nt.assert_equal(read_detected_enc, read_specified_enc)
|
||
|
|
assert "coding: iso-8859-5" in read_detected_enc
|
||
|
|
|
||
|
|
read_strip_enc_cookie = openpy.read_py_file(
|
||
|
|
nonascii_path, skip_encoding_cookie=True
|
||
|
|
)
|
||
|
|
assert "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("coding: iso-8859-5", source_no_cookie)
|