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