U
��_� c @ s� d dl Z d dlmZmZmZmZmZmZ d dlZd dl Z d dl
Z
d dl Z d dl Z G dd� de j
�ZG dd� d�ZG dd� dee j
�ZG d d
� d
ee j
�Zed kr�e �� dS ) � N)�TESTFN�unlink�unload�rmtree�
script_helper�captured_stdoutc @ s| e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd S ) �MiscSourceEncodingTestc C s( | � d�d�d� | � d�d�d� d S )Nu
Питон�utf-8s
Питонu \Пs \П)� assertEqual�encode��self� r
�1/usr/lib64/python3.8/test/test_source_encoding.py� test_pep263
s ��z"MiscSourceEncodingTest.test_pep263c C s. t ddd�}i }t||� | �|d d� d S )Ns
# coding: utf-8
u = 'ó'
�dummy�execZuu ó��compiler r �r �c�dr
r
r �test_compilestring s
z)MiscSourceEncodingTest.test_compilestringc
C sT zt ddd� W n6 tk
rF } z| �|j�d�d� W 5 d }~X Y n
X | �� d S )Ns # coding: cp932
print '�N'r r Z
u print '年')r � SyntaxErrorr Ztext�rstripZfail)r Zvr
r
r �test_issue2301 s
&z%MiscSourceEncodingTest.test_issue2301c C s. t ddd�}i }t||� | �|d d� d S )Nu # coding=latin-1
Æ = 'Æ'r r u Ær r r
r
r �test_issue4626& s
z%MiscSourceEncodingTest.test_issue4626c C sj t ddd�}i }t||� | �|d |d � | �t|d �t|d �� | �t|d �t|d �� d S )Nu a, b = '𐄏', '\U0001010F'r r �aZb)r r r ZlenZasciir r
r
r �test_issue3297, s
z%MiscSourceEncodingTest.test_issue3297c C s | � ttd� | � ttd� d S )Ns � s � )� assertRaisesr Zevalr r
r
r �test_issue78204 s z%MiscSourceEncodingTest.test_issue7820c C sR t jtjtj�tj�t�d�gt j d�}|�
� d }| � |j d� | �
d|� d S )Nzcoding20731.py)Zstderri r s SyntaxError)�
subprocessZPopen�sysZ
executable�os�path�join�dirname�__file__ZPIPEZ communicater Z
returncodeZ assertNotIn)r ZsubZerrr
r
r �
test_20731? s ��� z!MiscSourceEncodingTest.test_20731c C s� t ddd� t ddd� t ddd� | �td�� t ddd� W 5 Q R X | �td�� t d dd� W 5 Q R X | �td
�� t d dd� W 5 Q R X | �td�� t d dd� W 5 Q R X | �td
�� t d dd� W 5 Q R X d S ) Ns # -*- coding: iso-8859-15 -*-
r r s
s # -*- coding: utf-8 -*-
Zfakes # -*- coding: fake -*-
z iso-8859-15s! # -*- coding: iso-8859-15 -*-
ZBOMs # -*- coding: fake -*-
)r ZassertRaisesRegexr r r
r
r �test_error_messageH s& � �z)MiscSourceEncodingTest.test_error_messagec C � d}| � |� d S )NZ
bad_coding��verify_bad_module�r � module_namer
r
r �test_bad_codingY � z&MiscSourceEncodingTest.test_bad_codingc C r) )NZ bad_coding2r* r, r
r
r �test_bad_coding2] r/ z'MiscSourceEncodingTest.test_bad_coding2c C sd | � ttd| � tj�t�}tj�||d �}t|d��}|� � }W 5 Q R X | � tt
||d� d S )Nztest.�.pyZrbr ) r r �
__import__r" r# r% r& r$ �openZreadr )r r- r# �filename�fpZbytesr
r
r r+ a s z(MiscSourceEncodingTest.verify_bad_modulec C s" i }t d|� | �|d d� d S )Ns # coding: cp949
a = "��"
r u ぇ)r r )r r r
r
r �test_exec_valid_codingj s
z-MiscSourceEncodingTest.test_exec_valid_codingc C s� t t� td }t|ddd�}tj�dtj� zP|�2 |�
d � |�
d
� |�
d � |�
d d
� W 5 Q R X t � � t
t� W 5 tjd= t|� t|d � t|d � t t� t d� X d S )Nr1 ZwZcp1252)Zencodingr r ZoZ __pycache__z# -*- coding: cp1252 -*-
z'''A short string
z'''
z'A very long string %s'
A� XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX)r r r3 r! r# Zinsertr" Zcurdirr r �write� importlibZinvalidate_cachesr2 )r r4 Zfr
r
r �test_file_parseo s$
z&MiscSourceEncodingTest.test_file_parsec C sX d� d�}| �t��}t|dd� W 5 Q R X d}| j|jjd �|�|jjd d� d S )Nu # coding: ascii
☃r z<string>r zN'ascii' codec can't decode byte 0xe2 in position 16: ordinal not in range(128)r )Zmsg)r
r r r Z
assertTrueZ exceptionZargsZ
startswith)r Zinputr �expectedr
r
r �test_error_from_string� s
�z-MiscSourceEncodingTest.test_error_from_stringN)�__name__�
__module__� __qualname__r r r r r r r' r( r. r0 r+ r6 r9 r; r
r
r
r r s
r c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�AbstractSourceEncodingTestc C � d}| � |d� d S )Ns print(ascii("ä"))
� '\xe4'��check_script_output�r �srcr
r
r �test_default_coding� r/ z.AbstractSourceEncodingTest.test_default_codingc C r@ )Ns& #coding:iso8859-15
print(ascii("ä"))
� '\xc3\u20ac'rB rD r
r
r �test_first_coding_line� � z1AbstractSourceEncodingTest.test_first_coding_linec C r@ )Ns( #
#coding:iso8859-15
print(ascii("ä"))
rG rB rD r
r
r �test_second_coding_line� � z2AbstractSourceEncodingTest.test_second_coding_linec C r@ )Ns* #
#
#coding:iso8859-15
print(ascii("ä"))
rA rB rD r
r
r �test_third_coding_line� s z1AbstractSourceEncodingTest.test_third_coding_linec C r@ )Ns5 #coding:iso8859-15
#coding:latin1
print(ascii("ä"))
rG rB rD r
r
r �test_double_coding_line� s z2AbstractSourceEncodingTest.test_double_coding_linec C r@ )Ns4 #coding:iso8859-15 coding:latin1
print(ascii("ä"))
rG rB rD r
r
r � test_double_coding_same_line� rI z7AbstractSourceEncodingTest.test_double_coding_same_linec C r@ )Ns) #coding:iso-8859-15 �
print(ascii("ä"))
rG rB rD r
r
r � test_first_non_utf8_coding_line� rI z:AbstractSourceEncodingTest.test_first_non_utf8_coding_linec C r@ )Ns*
#coding:iso-8859-15 �
print(ascii("ä"))
rG rB rD r
r
r � test_second_non_utf8_coding_line� rK z;AbstractSourceEncodingTest.test_second_non_utf8_coding_linec C r@ )Ns print(ascii("ä"))
rA rB rD r
r
r �
test_utf8_bom� r/ z(AbstractSourceEncodingTest.test_utf8_bomc C r@ )Ns$ #coding:utf-8
print(ascii("ä"))
rA rB rD r
r
r �"test_utf8_bom_and_utf8_coding_line� rI z=AbstractSourceEncodingTest.test_utf8_bom_and_utf8_coding_lineN)
r< r= r> rF rH rJ rL rM rN rO rP rQ rR r
r
r
r r? � s r? c @ � e Zd Zdd� ZdS )�BytesSourceEncodingTestc C s< t � �}t|� W 5 Q R X |�� �d�}| �|�� |� d S )NZlatin1)r r Zgetvaluer
r r )r rE r: Zstdout�outr
r
r rC � s z+BytesSourceEncodingTest.check_script_outputN�r< r= r> rC r
r
r
r rT � � rT c @ rS )�FileSourceEncodingTestc
C sb t �� �>}tj�|d�}t|d��}|�|� W 5 Q R X t�|�}W 5 Q R X | � |j
� � |� d S )Nztest.pyZwb) �tempfileZTemporaryDirectoryr" r# r$ r3 r7 r Zassert_python_okr rU r )r rE r: ZtmpdZfnr5 Zresr
r
r rC � s
z*FileSourceEncodingTest.check_script_outputNrV r
r
r
r rX � rW rX Z__main__)ZunittestZ test.supportr r r r r r r8 r" r! r rY ZTestCaser r? rT rX r< Zmainr
r
r
r Z<module> s :