##// END OF EJS Templates
apply black on the file
Matthias Bussonnier -
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, '../../core/tests/nonascii.py')
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, 'rb') as f:
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, 'iso-8859-5')
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='iso-8859-5') as f:
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 'coding: iso-8859-5' in read_detected_enc
22 assert "coding: iso-8859-5" in read_detected_enc
21
23
22 read_strip_enc_cookie = openpy.read_py_file(nonascii_path, skip_encoding_cookie=True)
24 read_strip_enc_cookie = openpy.read_py_file(
23 assert 'coding: iso-8859-5' not in read_strip_enc_cookie
25 nonascii_path, skip_encoding_cookie=True
26 )
27 assert "coding: iso-8859-5" not in read_strip_enc_cookie
28
24
29
25 def test_source_to_unicode():
30 def test_source_to_unicode():
26 with io.open(nonascii_path, 'rb') as f:
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('coding: iso-8859-5', source_no_cookie)
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