OwlCyberSecurity - MANAGER
Edit File: test_exec_command.cpython-38.pyc
U �p�]; � @ 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�ZdS )� )�division�absolute_import�print_functionN)� 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 �I/usr/lib/python3/dist-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 2 s zemulate_nonposix.__init__c C s t j| _| jt _d S r )�os�name� _old_namer+ r r r r r 5 s zemulate_nonposix.__enter__c C s | j t_d S r )r. r, r- r r r r r 9 s zemulate_nonposix.__exit__N)r* r r r r r r) 0 s r) c C s� t t� ��: tt� ��$ tt�� t�d� W 5 Q R X W 5 Q R X W 5 Q R X tjdkr�t � �P t t� ��: tt� ��$ tt�� t�d� W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X d S )N�cd '.'�posix) r r r% r r �DeprecationWarningr r, r- r) r r r r �test_exec_command_stdout= s ( r2 c C s� t tdd���: tt� ��$ tt�� t�d� W 5 Q R X W 5 Q R X W 5 Q R X tjdkr�t � �P t t� ��: tt� ��$ tt�� t�d� W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X d S )Nzw+)�moder/ r0 ) r r r% r r r1 r r, r- r) r r r r �test_exec_command_stderrU s ( r4 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 �setupf s zTestExecCommand.setupc 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)"Zwin32)r r r6 �r �kws�s�or r r �check_nti s � zTestExecCommand.check_ntc K s� t j d|�\}}t|dk� t|dk� t j d|�\}}t|dk� t|dk� t j dddi|��\}}t|dk� t|dk� t j d|�\}}t|dk� t|dk� d tjk�rnd tjd < t j d|�\}}t|dk� t|d k� t j dd d i|��\}}t|dk� t|d k� t j d|�\}}t|dk� t|d k� tjd = t j d|�\}}t|dk� t|dk� t j d|�\}}t|dk� t|dk� t j d|�\}}t|dk� t|dk� t j d| j f|�\}}t|dk� t|dk� d S )N� echo Hellor �Hello� echo $AAAr8 � echo "$AAA"ZAAAZTereZBBBZHi� echo "$BBB"ZHey�this_is_not_a_command�echo path=$PATHz1"%s" -c "import sys,os;sys.stderr.write(os.name)"r0 )r> )r@ )rA )rA )rB )rB )rB )rB )rC )rD )r r r, �environr6 r9 r r r � check_posixs sT ��� zTestExecCommand.check_posixc G s� t j d| j f|�\}}t|dk� t|dk� t j d| j f|�\}}t|dk� t|dk� t j d| j f|�\}}t|dk� t|dk� t j d| j f|�\}}t|dk� t|d k� d S ) Nz "%s" -c "raise 'Ignore me.'"r r8 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 r6 r r9 r r r � check_basic� s< �� ��� �� �� zTestExecCommand.check_basicc K s� t � ��}d}tj�||�}t|d�}|�d� |�� tjd| j|f f|�\}}t |dk� t |dk� tjd| j|f fd|i|��\}}t |dk� t |dk� W 5 Q R X d S ) N�file�wr? z("%s" -c "f = open('%s', 'r'); f.close()"r r8 z9"%s" -c "f = open('%s', 'r'); print(f.read()); f.close()"Z execute_in) r r, �path�join�open�writer r r6 r )r r: Ztmpdir�fnZtmpfile�fr; r< r r r �check_execute_in� s4 ��� ���� z TestExecCommand.check_execute_inc C s� t t� ��� tt� ��x tt��d tjdkrF| jdd� | jdd� n"tjdkrh| jdd� | jdd� | j dd� | j dd� W 5 Q R X W 5 Q R X W 5 Q R X d S )Nr0 r )Zuse_tee� �nt) r r r% r r1 r, r- rF r= rQ r r r r � test_basic� s zTestExecCommand.test_basicN) r! r" r# r7 r= rF rH rQ rT r r r r r5 e s 4r5 )Z __future__r r r r, r Ztempfiler Znumpy.distutilsr Z numpy.distutils.exec_commandr Z numpy.testingr r r � version_info�ior �objectr r% r) r2 r4 r5 r r r r �<module> s