IPython/utils/tests/test_openpy.py

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)