File: //opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyo
�
� 9Zc @` sU d d l m Z m Z m Z d d l Z d d l Z d d l m Z m Z m Z d d l
m Z d d l m
Z
d d l m Z m Z m Z m Z d d l j j Z e j d d k r� d d l j Z d d l m Z d d l m Z n, d d l Z d d l m Z d d l m Z d d
� Z! d a" d � Z# d � Z$ d
Z% d Z& d Z' d Z( d d d d g Z) e
d � Z* d � Z+ d � Z, d � Z- d � Z. d � Z/ d � Z0 d � Z1 d � Z2 d e f d � � YZ3 d e f d! � � YZ4 d" e f d# � � YZ5 d$ e f d% � � YZ6 d& e f d' � � YZ7 d( e f d) � � YZ8 e9 d* k rQe � n d S(+ i ( t divisiont absolute_importt print_functionN( t mkdtempt mkstempt NamedTemporaryFile( t rmtree( t asbytes( t run_module_suitet TestCaset assert_t SkipTesti ( t urlparse( t URLErrorc C` s2 | t � k r" t d d � } | St d � � d S( s$ Stub to replace urlopen for testing.t prefixt urltmp_s Name or service not knownN( t
valid_httpurlR R
( t urlt datat tmpfile( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt urlopen_stub s c C` s t j a t t _ d S( N( t urllib_requestt urlopent old_urlopenR ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt setup$ s c C` s
t t _ d S( N( R R R ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt teardown+ s s http://www.google.com/s
index.htmls http://fake.abc.web/site/s fake.txts /etc/shadows ../../shadows
..\system.dats c:\windows\system.dats three is the magic numberc C` s8 t d d d d d | d t � \ } } t j | � | S( Nt suffixs .txtR t dstmp_t dirt text( R t Truet ost close( t filedirt fdt path( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt valid_textfile< s '
c C` s? t d d d d d | � \ } } t j | � t j | � | S( NR s .txtR R R ( R R R t remove( R! R" R# ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt invalid_textfileC s !
c C` s t t S( N( t http_patht http_file( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR K s c C` s t t S( N( t
http_fakepatht
http_fakefile( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt invalid_httpurlO s c C` s t S( N( R' ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt
valid_baseurlS s c C` s t S( N( R) ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt invalid_baseurlW s c C` s t S( N( R( ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt valid_httpfile[ s c C` s t S( N( R* ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt invalid_httpfile_ s t TestDataSourceOpenc B` sY e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
RS( c C` s% t � | _ t j | j � | _ d S( N( R t tmpdirt
datasourcet
DataSourcet ds( t self( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt setUpd s c C` s t | j � | ` d S( N( R R1 R4 ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt tearDownh s
c C` s- | j j t � � } t | � | j � d S( N( R4 t openR R
R ( R5 t fh( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_ValidHTTPl s
c C` sc t � } | j t | j j | � y | j j | � Wn& t k
r^ } t | j d k � n Xd S( N( R+ t assertRaisest IOErrorR4 R8 R
t errnot None( R5 R t e( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_InvalidHTTPq s c C` s | j t | j j t � � d S( N( R; R
R4 t _cacheR+ ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_InvalidHTTPCacheURLErrorz s c C` s9 t | j � } | j j | � } t | � | j � d S( N( R$ R1 R4 R8 R
R ( R5 t
local_fileR9 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_ValidFile} s
c C` s, t | j � } | j t | j j | � d S( N( R& R1 R; R< R4 R8 ( R5 t invalid_file( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_InvalidFile� s c C` s� y d d l } Wn t k
r) t � n Xt j j | j d � } | j | d � } | j t � | j
� | j j | � } | j � } | j
� | j
t | � d S( Ni s
foobar.txt.gzt w( t gzipt ImportErrorR R R# t joinR1 R8 t writet
magic_lineR R4 t readlinet assertEqual( R5 RH t filepatht fpt result( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_ValidGzipFile� s
c C` s� y d d l } Wn t k
r) t � n Xt j j | j d � } | j | d � } | j t � | j
� | j j | � } | j
� } | j
� | j t | � d S( Ni s foobar.txt.bz2RG ( t bz2RI R R R# RJ R1 t BZ2FileRK RL R R4 R8 RM RN ( R5 RS RO RP RQ ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_ValidBz2File� s
( t __name__t
__module__R6 R7 R: R@ RB RD RF RR RU ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR0 c s t TestDataSourceExistsc B` s> e Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C` s% t � | _ t j | j � | _ d S( N( R R1 R2 R3 R4 ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR6 � s c C` s t | j � | ` d S( N( R R1 R4 ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR7 � s
c C` s t | j j t � � � d S( N( R
R4 t existsR ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR: � s c C` s# | j | j j t � � t � d S( N( RN R4 RY R+ t False( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR@ � s c C` s^ t | j � } t | j j | � � t � } t | � } t | j j | � � t | � d S( N( R$ R1 R
R4 RY R R ( R5 R t localdir( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRD � s c C` s/ t | j � } | j | j j | � t � d S( N( R& R1 RN R4 RY RZ ( R5 R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRF � s ( RV RW R6 R7 R: R@ RD RF ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRX � s
t TestDataSourceAbspathc B` sP e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C` s1 t j j t � � | _ t j | j � | _ d S( N( R R# t abspathR R1 R2 R3 R4 ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR6 � s c C` s t | j � | ` d S( N( R R1 R4 ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR7 � s
c C` st t t � � \ } } } } } } t j j | j | | j t j � j d � � } | j | | j j
t � � � d S( Nt /( R R R R# RJ R1 t stript sepRN R4 R] ( R5 t schemet netloct upatht pmst qryt frgt
local_path( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR: � s ! c C` sa t | j � } t j j | � d } | j | | j j | � � | j | | j j | � � d S( Ni����( R$ R1 R R# t splitRN R4 R] ( R5 R t tmpfilename( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRD � s c C` st t t � � \ } } } } } } t j j | j | | j t j � j d � � } | j | | j j
t � � � d S( NR^ ( R R+ R R# RJ R1 R_ R` t assertNotEqualR4 R] R ( R5 Ra Rb Rc Rd Re Rf t invalidhttp( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR@ � s ! c C` sp t | j � } t | j � } t j j | � d } | j | | j j | � � | j | | j j | � � d S( Ni����( R$ R1 R R# Rh Rj R4 R] ( R5 t invalidfileR Ri ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRF � s
c ` s� t � j � } t j j | � d } � f d � } t | t � � j � j � � t | t � � j � j � � t | | � j � j � � t | | � j � j � � xJ t D]B } t | t
| � j � j � � t | | � j � j � � q� Wd S( Ni����c ` s t j j � j j | � � S( N( R R# R] R4 ( t x( R5 ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt <lambda>� s ( R$ R1 R R# Rh R
R t
startswithR+ t malicious_filesR' ( R5 R Ri t tmp_patht fn( ( R5 sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_sandboxing� s
c C` sY t j } z? d t _ | j � | j � | j � | j � | j � Wd | t _ Xd S( Ns \( R R` R: RD R@ RF Rs ( R5 t orig_os_sep( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_windows_os_sep� s
(
RV RW R6 R7 R: RD R@ RF Rs Ru ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR\ � s t TestRepositoryAbspathc B` s5 e Z d � Z d � Z d � Z d � Z d � Z RS( c C` s7 t j j t � � | _ t j t � | j � | _ d S( N( R R# R] R R1 R2 t
RepositoryR, t repos( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR6 s c C` s t | j � | ` d S( N( R R1 Rx ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR7 s
c C` s} t t � � \ } } } } } } t j j | j j | | j t j � j d � � } | j j t
� � } | j | | � d S( NR^ ( R R R R# RJ Rx t _destpathR_ R` R] R. RN ( R5 Ra Rb Rc Rd Re Rf Rg RO ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR: s
! c ` s � f d � } t | t � � j � j � � xJ t D]B } t | t | � j � j � � t | | � j � j � � q5 Wd S( Nc ` s t j j � j j | � � S( N( R R# R] Rx ( Rm ( R5 ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRn s ( R
R. Ro R1 Rp R' ( R5 Rq Rr ( ( R5 sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRs s
c C` s; t j } z! d t _ | j � | j � Wd | t _ Xd S( Ns \( R R` R: Rs ( R5 Rt ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRu s
( RV RW R6 R7 R: Rs Ru ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRv s
t TestRepositoryExistsc B` s> e Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C` s+ t � | _ t j t � | j � | _ d S( N( R R1 R2 Rw R, Rx ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR6 ) s c C` s t | j � | ` d S( N( R R1 Rx ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR7 - s
c C` s) t | j � } t | j j | � � d S( N( R$ R1 R
Rx RY ( R5 R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRD 1 s c C` s/ t | j � } | j | j j | � t � d S( N( R& R1 RN Rx RY RZ ( R5 R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRF 6 s c C` s t | j j t � � � d S( N( R
Rx RY R ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_RemoveHTTPFile: s c
C` sx t � } t | � \ } } } } } } t j j | j j | � } t j | d � t | � } t | j j
| � � d S( Ni� ( R R R R# RJ Rx Ry t mkdirR$ R
RY (
R5 t localfileRa Rb Rc Rd Re Rf Rg R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_CachedHTTPFile= s ( RV RW R6 R7 RD RF R{ R~ ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRz ( s t TestOpenFuncc B` s# e Z d � Z d � Z d � Z RS( c C` s t � | _ d S( N( R R1 ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR6 J s c C` s t | j � d S( N( R R1 ( R5 ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR7 M s c C` sb t | j � } t j | d | j �} t | � | j � t j | � } t | � | j � d S( Nt destpath( R$ R1 R2 R8 R
R ( R5 RC RP ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_DataSourceOpenP s
( RV RW R6 R7 R� ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR I s t __main__(: t
__future__R R R R t syst tempfileR R R t shutilR t numpy.compatR t
numpy.testingR R R
R t numpy.lib._datasourcet libt _datasourceR2 t version_infot urllib.requestt requestR t urllib.parseR t urllib.errorR
t urllib2R> R R R R R' R( R) R* Rp RL R$ R& R R+ R, R- R. R/ R0 RX R\ Rv Rz R RV ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt <module> sR " E A!!