| Current File : //usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyo |
�
E�`Qc @ s� 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 d d l Z d d l
Td d l m Z d d l
j j 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 � Z! d � Z" d � Z# d e$ f d � � YZ% d e$ f d � � YZ& d e$ f d! � � YZ' d" e$ f d# � � YZ( d$ e$ f d% � � YZ) d& e$ f d' � � YZ* e+ d( k r�e, � n d S() i����N( t mkdtempt mkstempt NamedTemporaryFile( t rmtree( t urlparse( t URLError( t *( t asbytesc 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( ( sF /usr/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 urllib2t urlopent old_urlopenR ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt setup s c C s
t t _ d S( N( R R R ( ( ( sF /usr/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( ( sF /usr/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 ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt invalid_textfile3 s !
c C s t t S( N( t http_patht http_file( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR
: s c C s t t S( N( t
http_fakepatht
http_fakefile( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt invalid_httpurl= s c C s t S( N( R! ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt
valid_baseurl@ s c C s t S( N( R# ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt invalid_baseurlC s c C s t S( N( R" ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt valid_httpfileF s c C s t S( N( R$ ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt invalid_httpfileI 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( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt setUpM s c C s t | j � | ` d S( N( R R+ R. ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt tearDownQ s
c C s- | j j t � � } t | � | j � d S( N( R. t openR
t assert_R ( R/ t fh( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_ValidHTTPU 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 IOErrorR. R2 R3 t errnot None( R/ R t e( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_InvalidHTTPZ s c C s | j t | j j t � � d S( N( R6 R R. t _cacheR% ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_InvalidHTTPCacheURLErrorc s c C s9 t | j � } | j j | � } t | � | j � d S( N( R R+ R. R2 R3 R ( R/ t
local_fileR4 ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_ValidFilef s
c C s, t | j � } | j t | j j | � d S( N( R R+ R6 R7 R. R2 ( R/ t invalid_file( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_InvalidFilel s c C s� y d d l } Wn& t k
r8 d d l } | j � 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 ImportErrort noset SkipTestR R t joinR+ R2 t writet
magic_lineR R. t readlinet assertEqual( R/ RC RE t filepatht fpt result( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_ValidGzipFilep s
c C s� y d d l } Wn& t k
r8 d d l } | j � n Xt j j | j d � } | j | d � } | j t
� | j � | j j
| � } | j � } | j � | j t
| � d S( Ni����s foobar.txt.bz2RB ( t bz2RD RE RF R R RG R+ t BZ2FileRH RI R R. R2 RJ RK ( R/ RP RE RL RM RN ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_ValidBz2File� s
( t __name__t
__module__R0 R1 R5 R; R= R? RA RO RR ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR* L 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 R+ R, R- R. ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR0 � s c C s t | j � | ` d S( N( R R+ R. ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR1 � s
c C s t | j j t � � � d S( N( R3 R. t existsR
( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR5 � s c C s# | j | j j t � � t � d S( N( RK R. RV R% t False( R/ ( ( sF /usr/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 R+ R3 R. RV R R ( R/ R
t localdir( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR? � s c C s/ t | j � } | j | j j | � t � d S( N( R R+ RK R. RV RW ( R/ R
( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRA � s ( RS RT R0 R1 R5 R; R? RA ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRU � 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 R+ R, R- R. ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR0 � s c C s t | j � | ` d S( N( R R+ R. ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR1 � 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 RG R+ t stript sepRK R. RZ ( R/ t schemet netloct upatht pmst qryt frgt
local_path( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR5 � s ! c C sq t | j � } t j j | � d } | j | | j j t j j | � d � � | j | | j j | � � d S( Ni����( R R+ R R t splitRK R. RZ ( R/ R
t tmpfilename( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR? � 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 RG R+ R\ R] t assertNotEqualR. RZ R
( R/ R^ R_ R` Ra Rb Rc t invalidhttp( ( sF /usr/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 R+ R R Re Rg R. RZ ( R/ t invalidfileR
Rf ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRA � 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 RZ R. ( t x( R/ ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt <lambda>� s ( R R+ R R Re R3 R
t
startswithR% t malicious_filesR! ( R/ R
Rf t tmp_patht fn( ( R/ sF /usr/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] R5 R? R; RA Rp ( R/ t orig_os_sep( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_windows_os_sep� s
(
RS RT R0 R1 R5 R? R; RA Rp Rr ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRY � 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 RZ R R+ R, t
RepositoryR& t repos( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR0 � s c C s t | j � | ` d S( N( R R+ Ru ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR1 � 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 RG Ru t _destpathR\ R] RZ R( RK ( R/ R^ R_ R` Ra Rb Rc Rd RL ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR5 � 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 RZ Ru ( Rj ( R/ ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRk s ( R3 R( Rl R+ Rm R! ( R/ Rn Ro ( ( R/ sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRp s
c C s; t j } z! d t _ | j � | j � Wd | t _ Xd S( Ns \( R R] R5 Rp ( R/ Rq ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRr s
( RS RT R0 R1 R5 Rp Rr ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRs � 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 R+ R, Rt R& Ru ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR0 s c C s t | j � | ` d S( N( R R+ Ru ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR1 s
c C s) t | j � } t | j j | � � d S( N( R R+ R3 Ru RV ( R/ R
( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR? s c C s/ t | j � } | j | j j | � t � d S( N( R R+ RK Ru RV RW ( R/ R
( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRA ! s c C s t | j j t � � � d S( N( R3 Ru RV R
( R/ ( ( sF /usr/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 RG Ru Rv t mkdirR R3 RV (
R/ t localfileR^ R_ R` Ra Rb Rc Rd R
( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_CachedHTTPFile( s ( RS RT R0 R1 R? RA Rx R{ ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyRw s t TestOpenFuncc B s# e Z d � Z d � Z d � Z RS( c C s t � | _ d S( N( R R+ ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR0 4 s c C s t | j � d S( N( R R+ ( R/ ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR1 7 s c C sb t | j � } t j | d | j �} t | � | j � t j | � } t | � | j � d S( Nt destpath( R R+ R, R2 R3 R ( R/ R> RM ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt test_DataSourceOpen: s
( RS RT R0 R1 R~ ( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyR| 3 s t __main__(- R t tempfileR R R t shutilR R R R t
numpy.testingt numpy.compatR t numpy.lib._datasourcet libt _datasourceR, R9 R R R R R! R" R# R$ Rm RI R R R
R% R& R' R( R) t TestCaseR* RU RY Rs Rw R| RS t run_module_suite( ( ( sF /usr/lib64/python2.7/site-packages/numpy/lib/tests/test__datasource.pyt <module> sF
G A!