Edit File: test_exec_command.cpython-37.pyc
B ��FdU � @ s� d dl mZmZmZ d dlZd dlZd dlmZ d dlm Z d dl m Z d dl m Z mZmZ ejd dkrxd dlmZ n d dlmZ G d d � d e�ZG d d � d e�ZG d d� de�Zdd� Zdd� ZG dd� de �Zedkr�e� dS )� )�division�absolute_import�print_functionN)� TemporaryFile)� exec_command)� get_pythonexe)�TestCase�run_module_suite�tempdir� )�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 ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/distutils/tests/test_exec_command.py�__init__ s zredirect_stdout.__init__c C s t j| _| jt _d S )N)r r � old_stdoutr )r r r r � __enter__ s zredirect_stdout.__enter__c C s | j �� | jt_| j �� d S )N)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 )N)r �stderr�_stderr)r r" r r r r # s zredirect_stderr.__init__c C s t j| _| jt _d S )N)r r" � old_stderrr# )r r r r r & s zredirect_stderr.__enter__c C s | j �� | jt_| j �� d S )N)r# r r$ r r" r )r r r r r r r r * s zredirect_stderr.__exit__)N)r r r r r r r 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 )N)� _new_name)r �osnamer r r r 2 s zemulate_nonposix.__init__c C s t j| _| jt _d S )N)�os�name� _old_namer' )r r r r r 5 s zemulate_nonposix.__enter__c C s | j t_d S )N)r+ r) r* )r r r r r r r r 9 s zemulate_nonposix.__exit__N)r& )r r r r r r r r r r r r% 0 s r% c C s� t t� ��& tt� �� t�d� W d Q R X W d Q R X tjdkr�t� �< t t� ��& tt� �� t�d� W d Q R X W d Q R X W d Q R X d S )Nzcd '.'�posix)r r r! r r r) r* r% r r r r �test_exec_command_stdout= s r- c C s� t tdd���& tt� �� t�d� W d Q R X W d Q R X tjdkr�t� �< t t� ��& tt� �� t�d� W d Q R X W d Q R X W d Q R X d S )Nzw+)�modezcd '.'r, )r r r! r r r) r* r% r r r r �test_exec_command_stderrS s r/ c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestExecCommandc C s t � | _d S )N)r �pyexe)r r r r �setUpb s zTestExecCommand.setUpc K sV t � d�\}}| �|d� | �|d� t � d| j �\}}| �|d� | �|d� d S )Nzcmd /C echo path=%path%r � z3"%s" -c "import sys;sys.stderr.write(sys.platform)"�win32)r � assertEqual�assertNotEqualr1 )r �kws�s�or r r �check_nte s zTestExecCommand.check_ntc K s� t j d|�\}}| �|d� | �|d� t j d|�\}}| �|d� | �|d� t j dddi|��\}}| �|d� | �|d� t j d|�\}}| �|d� | �|d� d tjk�rnd tjd < t j d|�\}}| �|d� | �|d � t j dd d i|��\}}| �|d� | �|d � t j d|�\}}| �|d� | �|d � tjd = t j d|�\}}| �|d� | �|d� t j d|�\}}| �|d� | �|d� t j d|�\}}| �|d� | �|d� t j d| j f|�\}}| �|d� | �|d� d S )N� echo Hellor �Hello� echo $AAAr3 � echo "$AAA"ZAAAZTereZBBB�Hi� 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@ )rA )rB )r r5 r) �environr6 r1 )r r7 r8 r9 r r r � check_posixo sL zTestExecCommand.check_posixc G s� t j d| j f|�\}}| �|d� | �|d� t j d| j f|�\}}| �|d� | �|d� t j d| j f|�\}}| �|d� | �|d� t j d| j f|�\}}| �|d� | �|d � d S ) Nz "%s" -c "raise 'Ignore me.'"r r3 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 r1 r6 r5 )r r7 r8 r9 r r r � check_basic� s" zTestExecCommand.check_basicc K s� t � ��}d}tj�||�}t|d�}|�d� |�� tjd| j|f f|�\}}| � |d� | � |d� tjd| j|f fd|i|��\}}| � |d� | � |d� W d Q R X d S ) N�file�wr< z("%s" -c "f = open('%s', 'r'); f.close()"r r3 z9"%s" -c "f = open('%s', 'r'); print(f.read()); f.close()"� execute_in) r r) �path�join�open�writer r r1 r6 r5 )r r7 �tmpdir�fn�tmpfile�fr8 r9 r r r �check_execute_in� s z TestExecCommand.check_execute_inc C s� t t� ��z tt� ��d tjdkr<| jdd� | jdd� n"tjdkr^| jdd� | jdd� | jdd� | jdd� W d Q R X W d Q R X d S )Nr, r )�use_tee� �nt)r r r! r) r* rD r: rR )r r r r � test_basic� s zTestExecCommand.test_basicN) r r r r2 r: rD rF rR rV r r r r r0 a s 4r0 �__main__)� __future__r r r r) r �tempfiler �numpy.distutilsr � numpy.distutils.exec_commandr Z numpy.testingr r r � version_info�ior �objectr r! r% r- r/ r0 r r r r r �<module> s" y