Edit File: test__datasource.cpython-37.pyc
B ��Fd�( � @ s� d dl mZmZmZ d dlZd dlZd dlmZmZm Z d dl m Z d dl m Z mZmZmZ d dlm mZ ejd dkr�d dlmZ d dlmZ d dlm Z n d dl Zd dlmZ d dl m Z d5d d �Z da d d � Z d d� Z!dZ"dZ#dZ$dZ%ddddgZ&dZ'dd� Z(dd� Z)d d � Z*d d � Z+d d!� Z,d"d#� Z-d$d%� Z.d&d'� Z/G d(d)� d)e�Z0G d*d+� d+e�Z1G d,d-� d-e�Z2G d.d/� d/e�Z3G d0d1� d1e�Z4G d2d3� d3e�Z5e6d4k�r�e � dS )6� )�division�absolute_import�print_functionN)�mkdtemp�mkstemp�NamedTemporaryFile)�rmtree)�run_module_suite�TestCase�assert_�SkipTest� )�urlparse)�URLErrorc C s$ | t � krtdd�}|S td��dS )z$Stub to replace urlopen for testing.Zurltmp_)�prefixzName or service not knownN)� valid_httpurlr r )�url�data�tmpfile� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test__datasource.py� urlopen_stub s r c C s t jatt _d S )N)�urllib_request�urlopen� old_urlopenr r r r r �setup# s r c C s t t_d S )N)r r r r r r r �teardown* s r zhttp://www.google.com/z index.htmlzhttp://fake.abc.web/site/zfake.txtz /etc/shadowz ../../shadowz ..\system.datzc:\windows\system.dats three is the magic numberc C s" t dd| dd�\}}t�|� |S )Nz.txt�dstmp_T)�suffixr �dir�text)r �os�close)�filedir�fd�pathr r r �valid_textfile; s r& c C s* t dd| d�\}}t�|� t�|� |S )Nz.txtr )r r r )r r! r" �remove)r# r$ r% r r r �invalid_textfileB s r( c C s t t S )N)� http_path� http_filer r r r r J s r c C s t t S )N)� http_fakepath� http_fakefiler r r r �invalid_httpurlN s r- c C s t S )N)r) r r r r � valid_baseurlR s r. c C s t S )N)r+ r r r r �invalid_baseurlV s r/ c C s t S )N)r* r r r r �valid_httpfileZ s r0 c C s t S )N)r, r r r r �invalid_httpfile^ s r1 c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�TestDataSourceOpenc C s t � | _t�| j�| _d S )N)r �tmpdir� datasource� DataSource�ds)�selfr r r �setUpc s zTestDataSourceOpen.setUpc C s t | j� | `d S )N)r r3 r6 )r7 r r r �tearDowng s zTestDataSourceOpen.tearDownc C s" | j �t� �}t|� |�� d S )N)r6 �openr r r" )r7 �fhr r r �test_ValidHTTPk s z!TestDataSourceOpen.test_ValidHTTPc C s^ t � }| �t| jj|� y| j�|� W n0 tk rX } zt|jd k� W d d }~X Y nX d S )N)r- � assertRaises�IOErrorr6 r: r �errno)r7 r �er r r �test_InvalidHTTPp s z#TestDataSourceOpen.test_InvalidHTTPc C s | � t| jjt� � d S )N)r= r r6 �_cacher- )r7 r r r � test_InvalidHTTPCacheURLErrory s z0TestDataSourceOpen.test_InvalidHTTPCacheURLErrorc C s* t | j�}| j�|�}t|� |�� d S )N)r& r3 r6 r: r r" )r7 � local_filer; r r r �test_ValidFile| s z!TestDataSourceOpen.test_ValidFilec C s t | j�}| �t| jj|� d S )N)r( r3 r= r> r6 r: )r7 Z invalid_filer r r �test_InvalidFile� s z#TestDataSourceOpen.test_InvalidFilec C s� y dd l }W n tk r$ t�Y nX tj�| jd�}|�|d�}|�t � |� � | j �|�}|� � }|� � | � t |� d S )Nr z foobar.txt.gz�w)�gzip� ImportErrorr r! r% �joinr3 r: �write� magic_liner" r6 �readline� assertEqual)r7 rH �filepath�fp�resultr r r �test_ValidGzipFile� s z%TestDataSourceOpen.test_ValidGzipFilec C s� y dd l }W n tk r$ t�Y nX tj�| jd�}|�|d�}|�t � |� � | j � |�}|� � }|� � | �t |� d S )Nr zfoobar.txt.bz2rG )�bz2rI r r! r% rJ r3 �BZ2FilerK rL r" r6 r: rM rN )r7 rS rO rP rQ r r r �test_ValidBz2File� s z$TestDataSourceOpen.test_ValidBz2FileN) �__name__� __module__� __qualname__r8 r9 r< rA rC rE rF rR rU r r r r r2 b s r2 c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestDataSourceExistsc C s t � | _t�| j�| _d S )N)r r3 r4 r5 r6 )r7 r r r r8 � s zTestDataSourceExists.setUpc C s t | j� | `d S )N)r r3 r6 )r7 r r r r9 � s z TestDataSourceExists.tearDownc C s t | j�t� �� d S )N)r r6 �existsr )r7 r r r r< � s z#TestDataSourceExists.test_ValidHTTPc C s | � | j�t� �d� d S )NF)rN r6 rZ r- )r7 r r r rA � s z%TestDataSourceExists.test_InvalidHTTPc C sD t | j�}t| j�|�� t� }t |�}t| j�|�� t|� d S )N)r& r3 r r6 rZ r r )r7 r Zlocaldirr r r rE � s z#TestDataSourceExists.test_ValidFilec C s" t | j�}| �| j�|�d� d S )NF)r( r3 rN r6 rZ )r7 r r r r rF � s z%TestDataSourceExists.test_InvalidFileN) rV rW rX r8 r9 r< rA rE rF r r r r rY � s rY c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestDataSourceAbspathc C s" t j�t� �| _t�| j�| _d S )N)r! r% �abspathr r3 r4 r5 r6 )r7 r r r r8 � s zTestDataSourceAbspath.setUpc C s t | j� | `d S )N)r r3 r6 )r7 r r r r9 � s z TestDataSourceAbspath.tearDownc C sP t t� �\}}}}}}tj�| j||�tj��d��}| �|| j � t� �� d S )N�/) r r r! r% rJ r3 �strip�seprN r6 r\ )r7 �scheme�netloc�upath�pms�qry�frg� local_pathr r r r< � s z$TestDataSourceAbspath.test_ValidHTTPc C sF t | j�}tj�|�d }| �|| j�|�� | �|| j�|�� d S )N���)r& r3 r! r% �splitrN r6 r\ )r7 r � tmpfilenamer r r rE � s z$TestDataSourceAbspath.test_ValidFilec C sP t t� �\}}}}}}tj�| j||�tj��d��}| �|| j � t � �� d S )Nr] ) r r- r! r% rJ r3 r^ r_ �assertNotEqualr6 r\ r )r7 r` ra rb rc rd re Z invalidhttpr r r rA � s z&TestDataSourceAbspath.test_InvalidHTTPc C sP t | j�}t | j�}tj�|�d }| �|| j�|�� | �|| j�|�� d S )Nrg )r& r3 r! r% rh rj r6 r\ )r7 Z invalidfiler ri r r r rF � s z&TestDataSourceAbspath.test_InvalidFilec s� t � j�}tj�|�d }� fdd�}t|t� ��� j�� t|t� ��� j�� t||��� j�� t||��� j�� x8t D ]0}t|t | ��� j�� t||��� j�� q�W d S )Nrg c s t j�� j�| ��S )N)r! r% r\ r6 )�x)r7 r r �<lambda>� � z7TestDataSourceAbspath.test_sandboxing.<locals>.<lambda>) r& r3 r! r% rh r r � startswithr- �malicious_filesr) )r7 r ri �tmp_path�fnr )r7 r �test_sandboxing� s z%TestDataSourceAbspath.test_sandboxingc C sF t j}z2dt _| �� | �� | �� | �� | �� W d |t _X d S )N�\)r! r_ r< rE rA rF rr )r7 � orig_os_sepr r r �test_windows_os_sep� s z)TestDataSourceAbspath.test_windows_os_sepN) rV rW rX r8 r9 r< rE rA rF rr ru r r r r r[ � s r[ c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestRepositoryAbspathc C s& t j�t� �| _t�t� | j�| _d S )N) r! r% r\ r r3 r4 � Repositoryr. �repos)r7 r r r r8 s zTestRepositoryAbspath.setUpc C s t | j� | `d S )N)r r3 rx )r7 r r r r9 s z TestRepositoryAbspath.tearDownc C sV t t� �\}}}}}}tj�| jj||�tj��d��}| j� t � �}| � ||� d S )Nr] ) r r r! r% rJ rx � _destpathr^ r_ r\ r0 rN ) r7 r` ra rb rc rd re rf rO r r r r< s z$TestRepositoryAbspath.test_ValidHTTPc s` � fdd�}t |t� ��� j�� x8tD ]0}t |t| ��� j�� t ||��� j�� q(W d S )Nc s t j�� j�| ��S )N)r! r% r\ rx )rk )r7 r r rl rm z7TestRepositoryAbspath.test_sandboxing.<locals>.<lambda>)r r0 rn r3 ro r) )r7 rp rq r )r7 r rr s z%TestRepositoryAbspath.test_sandboxingc C s. t j}zdt _| �� | �� W d |t _X d S )Nrs )r! r_ r< rr )r7 rt r r r ru s z)TestRepositoryAbspath.test_windows_os_sepN)rV rW rX r8 r9 r< rr ru r r r r rv s rv c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestRepositoryExistsc C s t � | _t�t� | j�| _d S )N)r r3 r4 rw r. rx )r7 r r r r8 ( s zTestRepositoryExists.setUpc C s t | j� | `d S )N)r r3 rx )r7 r r r r9 , s z TestRepositoryExists.tearDownc C s t | j�}t| j�|�� d S )N)r& r3 r rx rZ )r7 r r r r rE 0 s z#TestRepositoryExists.test_ValidFilec C s" t | j�}| �| j�|�d� d S )NF)r( r3 rN rx rZ )r7 r r r r rF 5 s z%TestRepositoryExists.test_InvalidFilec C s t | j�t� �� d S )N)r rx rZ r )r7 r r r �test_RemoveHTTPFile9 s z(TestRepositoryExists.test_RemoveHTTPFilec C sT t � }t|�\}}}}}}tj�| jj|�}t�|d� t|�} t | j� | �� d S )Ni� ) r r r! r% rJ rx ry �mkdirr& r rZ ) r7 � localfiler` ra rb rc rd re rf r r r r �test_CachedHTTPFile< s z(TestRepositoryExists.test_CachedHTTPFileN) rV rW rX r8 r9 rE rF r{ r~ r r r r rz ' s rz c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestOpenFuncc C s t � | _d S )N)r r3 )r7 r r r r8 I s zTestOpenFunc.setUpc C s t | j� d S )N)r r3 )r7 r r r r9 L s zTestOpenFunc.tearDownc C sH t | j�}tj|| jd�}t|� |�� t�|�}t|� |�� d S )N)�destpath)r& r3 r4 r: r r" )r7 rD rP r r r �test_DataSourceOpenO s z TestOpenFunc.test_DataSourceOpenN)rV rW rX r8 r9 r� r r r r r H s r �__main__)N)7� __future__r r r r! �sys�tempfiler r r �shutilr Z numpy.testingr r r r Znumpy.lib._datasource�lib� _datasourcer4 � version_info�urllib.request�requestr � urllib.parser � urllib.errorr �urllib2r r r r r) r* r+ r, ro rL r& r( r r- r. r/ r0 r1 r2 rY r[ rv rz r rV r r r r �<module> sP E A!!