FilesExpand file tree

 master

/

test_openpy.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

39 lines (31 loc) · 1.41 KB

 master

/

test_openpy.py

Copy path

Top

File metadata and controls

• Code
• Blame

39 lines (31 loc) · 1.41 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

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():

f = open(nonascii_path, 'rb')

enc, lines = openpy.detect_encoding(f.readline)

nt.assert_equal(enc, 'iso-8859-5')

def test_read_file():

read_specified_enc = io.open(nonascii_path, encoding='iso-8859-5').read()

read_detected_enc = openpy.read_py_file(nonascii_path, skip_encoding_cookie=False)

nt.assert_equal(read_detected_enc, read_specified_enc)

assert u'coding: iso-8859-5' in read_detected_enc

read_strip_enc_cookie = openpy.read_py_file(nonascii_path, skip_encoding_cookie=True)

assert u'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(u'coding: iso-8859-5', source_no_cookie)

def test_list_readline():

l = ['a', 'b']

readline = openpy._list_readline(l)

nt.assert_equal(readline(), 'a')

nt.assert_equal(readline(), 'b')

with nt.assert_raises(StopIteration):

readline()