Current File : //lib64/python3.9/site-packages/numpy/distutils/tests/__pycache__/test_exec_command.cpython-39.pyc
a
z[yc� � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z m
Z
m Z d dl m
Z
G dd� d�ZG d d
� d
�ZG d d � d �Zd
d� Zdd� ZG dd� d�ZdS )� N)�
TemporaryFile)� exec_command)�
get_pythonexe)�tempdir�assert_� assert_warns)�StringIOc @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
�redirect_stdoutz9Context manager to redirect stdout for exec_command test.Nc C s |pt j| _d S �N)�sys�stdout�_stdout)�selfr � r �M/usr/lib64/python3.9/site-packages/numpy/distutils/tests/test_exec_command.py�__init__ s zredirect_stdout.__init__c C s t j| _| jt _d S r
)r r �
old_stdoutr
�r r r r � __enter__ s zredirect_stdout.__enter__c C s | j �� | jt_| j �� d S r
)r
�flushr r r �close�r �exc_type� exc_value� tracebackr r r �__exit__ s
zredirect_stdout.__exit__)N��__name__�
__module__� __qualname__�__doc__r r r r r r r r
s
r c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
�redirect_stderrz9Context manager to redirect stderr for exec_command test.Nc C s |pt j| _d S r
)r �stderr�_stderr)r r" r r r r s zredirect_stderr.__init__c C s t j| _| jt _d S r
)r r" �
old_stderrr# r r r r r ! s zredirect_stderr.__enter__c C s | j �� | jt_| j �� d S r
)r# r r$ r r" r r r r r r % s
zredirect_stderr.__exit__)Nr r r r r r! s
r! c @ s* e Zd ZdZd
dd�Zdd� Zdd� Zd S ) �emulate_nonposixz.Context manager to emulate os.name != 'posix' � non-posixc C s
|| _ d S r
)� _new_name)r Zosnamer r r r - s zemulate_nonposix.__init__c C s t j| _| jt _d S r
)�os�name� _old_namer' r r r r r 0 s zemulate_nonposix.__enter__c C s | j t_d S r
)r* r( r) r r r r r 4 s zemulate_nonposix.__exit__N)r& r r r r r r% + s
r% c C sF t t� ��l tt� ��B tt�� t�d� W d � n1 s@0 Y W d � n1 s^0 Y W d � n1 s|0 Y tjdk�rBt � �� t t� ��l tt� ��B tt�� t�d� W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 �s0 Y W d � n1 �s80 Y d S )N�cd '.'�posix)
r r r! r r �DeprecationWarningr r( r) r% r r r r �test_exec_command_stdout8 s
d
r. c C sJ t tdd���l tt� ��B tt�� t�d� W d � n1 sD0 Y W d � n1 sb0 Y W d � n1 s�0 Y tjdk�rFt � �� t t� ��l tt� ��B tt�� t�d� W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 �s 0 Y W d � n1 �s<0 Y d S )Nzw+)�moder+ r, )
r r r! r r r- r r( r) r% r r r r �test_exec_command_stderrP s
d
r0 c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestExecCommandc C s t � | _d S r
)r �pyexer r r r � setup_methoda s z TestExecCommand.setup_methodc K sV t � d�\}}t|dk� t|dk� t � d| j �\}}t|dk� t|dk� d S )Nzcmd /C echo path=%path%r � z3"%s" -c "import sys;sys.stderr.write(sys.platform)"�win32)r r r2 �r �kws�s�or r r �check_ntd s � zTestExecCommand.check_ntc K s t j di |��\}}t|dk� t|dk� t j di |��\}}t|dk� t|dk� t j dddi|��\}}t|dk� t|dk� t j di |��\}}t|dk� t|dk� d tjv�r�d
tjd < t j di |��\}}t|dk� t|d
k� t j dd d i|��\}}t|dk� t|d k� t j di |��\}}t|dk� t|d
k� tjd = t j di |��\}}t|dk� t|dk� t j di |��\}}t|dk� t|dk� t j di |��\}}t|dk� t|dk� t j d| j fi |��\}}t|dk� t|dk� d S )N�
echo Hellor �Hello� echo $AAAr4 � echo "$AAA"ZAAAZTereZBBBZHi� echo "$BBB"ZHey�this_is_not_a_command�echo path=$PATHz1"%s" -c "import sys,os;sys.stderr.write(os.name)"r, )r; )r= )r> )r> )r? )r? )r? )r? )r@ )rA )r r r( �environr2 r6 r r r � check_posixn sT
���
zTestExecCommand.check_posixc G s� t j d| j fi |��\}}t|dk� t|dk� t j d| j fi |��\}}t|dk� t|dk� t j d| j fi |��\}}t|dk� t|dk� t j d| j fi |��\}}t|dk� t|d k� d S )
Nz "%s" -c "raise 'Ignore me.'"r r4 zV"%s" -c "import sys;sys.stderr.write('0');sys.stderr.write('1');sys.stderr.write('2')"Z012z!"%s" -c "import sys;sys.exit(15)"� z"%s" -c "print('Heipa'")ZHeipa)r r2 r r6 r r r � check_basic� s< ��
���
��
��
zTestExecCommand.check_basicc K s� t � ��}d}tj�||�}t|d��}|�d� W d � n1 sD0 Y tjd| j|f fi |��\}}t|dk� t|dk� tjd| j|f fd|i|��\}}t|dk� t|dk� W d � n1 s�0 Y d S ) N�file�wr< z("%s" -c "f = open('%s', 'r'); f.close()"r r4 z9"%s" -c "f = open('%s', 'r'); print(f.read()); f.close()"Z
execute_in) r r( �path�join�open�writer r2 r )r r7 Ztmpdir�fnZtmpfile�fr8 r9 r r r �check_execute_in� s2 (���
����
z TestExecCommand.check_execute_inc
C s� t t� ��� tt� ��� tt��n tjdkrF| jdd� | jdd� n"tjdkrh| jdd� | jdd� | j dd� | j dd� W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y d S )Nr, r )Zuse_tee� �nt)
r r r! r r- r( r) rC r: rN r r r r �
test_basic� s
zTestExecCommand.test_basicN) r r r r3 r: rC rE rN rQ r r r r r1 ` s
4r1 )r( r Ztempfiler Znumpy.distutilsr Z numpy.distutils.exec_commandr Z
numpy.testingr r r �ior r r! r% r. r0 r1 r r r r �<module> s
Mini Shell