Show More
@@ -1,32 +1,39 b'' | |||||
1 | import io |
|
1 | import io | |
2 | import os.path |
|
2 | import os.path | |
3 | import nose.tools as nt |
|
3 | import nose.tools as nt | |
4 |
|
4 | |||
5 | from IPython.utils import openpy |
|
5 | from IPython.utils import openpy | |
6 |
|
6 | |||
7 | mydir = os.path.dirname(__file__) |
|
7 | mydir = os.path.dirname(__file__) | |
8 |
nonascii_path = os.path.join(mydir, |
|
8 | nonascii_path = os.path.join(mydir, "../../core/tests/nonascii.py") | |
|
9 | ||||
9 |
|
10 | |||
10 | def test_detect_encoding(): |
|
11 | def test_detect_encoding(): | |
11 |
with open(nonascii_path, |
|
12 | with open(nonascii_path, "rb") as f: | |
12 | enc, lines = openpy.detect_encoding(f.readline) |
|
13 | enc, lines = openpy.detect_encoding(f.readline) | |
13 |
nt.assert_equal(enc, |
|
14 | nt.assert_equal(enc, "iso-8859-5") | |
|
15 | ||||
14 |
|
16 | |||
15 | def test_read_file(): |
|
17 | def test_read_file(): | |
16 |
with io.open(nonascii_path, encoding= |
|
18 | with io.open(nonascii_path, encoding="iso-8859-5") as f: | |
17 | read_specified_enc = f.read() |
|
19 | read_specified_enc = f.read() | |
18 | read_detected_enc = openpy.read_py_file(nonascii_path, skip_encoding_cookie=False) |
|
20 | read_detected_enc = openpy.read_py_file(nonascii_path, skip_encoding_cookie=False) | |
19 | nt.assert_equal(read_detected_enc, read_specified_enc) |
|
21 | nt.assert_equal(read_detected_enc, read_specified_enc) | |
20 |
assert |
|
22 | assert "coding: iso-8859-5" in read_detected_enc | |
|
23 | ||||
|
24 | read_strip_enc_cookie = openpy.read_py_file( | |||
|
25 | nonascii_path, skip_encoding_cookie=True | |||
|
26 | ) | |||
|
27 | assert "coding: iso-8859-5" not in read_strip_enc_cookie | |||
21 |
|
28 | |||
22 | read_strip_enc_cookie = openpy.read_py_file(nonascii_path, skip_encoding_cookie=True) |
|
|||
23 | assert 'coding: iso-8859-5' not in read_strip_enc_cookie |
|
|||
24 |
|
29 | |||
25 | def test_source_to_unicode(): |
|
30 | def test_source_to_unicode(): | |
26 |
with io.open(nonascii_path, |
|
31 | with io.open(nonascii_path, "rb") as f: | |
27 | source_bytes = f.read() |
|
32 | source_bytes = f.read() | |
28 | nt.assert_equal(openpy.source_to_unicode(source_bytes, skip_encoding_cookie=False).splitlines(), |
|
33 | nt.assert_equal( | |
29 | source_bytes.decode('iso-8859-5').splitlines()) |
|
34 | openpy.source_to_unicode(source_bytes, skip_encoding_cookie=False).splitlines(), | |
|
35 | source_bytes.decode("iso-8859-5").splitlines(), | |||
|
36 | ) | |||
30 |
|
37 | |||
31 | source_no_cookie = openpy.source_to_unicode(source_bytes, skip_encoding_cookie=True) |
|
38 | source_no_cookie = openpy.source_to_unicode(source_bytes, skip_encoding_cookie=True) | |
32 |
nt.assert_not_in( |
|
39 | nt.assert_not_in("coding: iso-8859-5", source_no_cookie) |
General Comments 0
You need to be logged in to leave comments.
Login now