lnT{ Z d dl Z d dlZd dlmZmZmZ d dlmZ d dlm c m
Z d dl m
Z
mZmZ d dlmZ d dlmZ d dlmZ d%dZdad Zd Zd
Zd Z d Z d
Z g dZ dZ d Z!d Z"d Z#d Z$d Z%d Z&d Z'd Z( G d d Z) G d d Z* G d d Z+ G d d Z, G d d! Z- G d" d# Z.d$ Z/dS )& N)mkdtempmkstempNamedTemporaryFile)rmtree)assert_ assert_equal
assert_raises)urlparse)URLErrorc h | t k rt d }|S t d )z$Stub to replace urlopen for testing.urltmp_)prefixzName or service not known)
valid_httpurlr r )urldatatmpfiles t/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test__datasource.py urlopen_stubr s5
moo $I6662333 c @ t j at t _ d S N)urllib_requesturlopen old_urlopenr r r setup_moduler s !(K )Nr c ( t t _ d S r )r r r r r r teardown_moduler ! s (Nr zhttp://www.google.com/z
index.htmlzhttp://fake.abc.web/site/zfake.txt)z /etc/shadowz ../../shadowz
..\system.datzc:\windows\system.dats three is the magic numberc Z t dd| d \ }}t j | |S )N.txtdstmp_T)suffixr dirtext)r osclosefiledirfdpaths r valid_textfiler+ 2 s. fX7NNNHBHRLLL Kr c t dd| \ }}t j | t j | |S )Nr r! )r" r r# )r r% r& remover' s r invalid_textfiler. 9 s9 fX7CCCHBHRLLLIdOOO Kr c t t z S r ) http_path http_filer r r r r A s
Y r c t t z S r )
http_fakepath
http_fakefiler r r invalid_httpurlr5 E s
&&r c t S r )r0 r r r
valid_baseurlr7 I r c t S r )r3 r r r invalid_baseurlr: M r c t S r )r1 r r r valid_httpfiler= Q r8 r c t S r )r4 r r r invalid_httpfiler? U r; r c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
S ) TestDataSourceOpenc h t | _ t j | j | _ d S r r tmpdir
datasource
DataSourcedsselfs r setup_methodz TestDataSourceOpen.setup_methodZ % ii ' 44r c 2 t | j | `d S r r rD rG rH s r teardown_methodz"TestDataSourceOpen.teardown_method^ t{ GGGr c | j t }t | | d S r )rG openr r r& )rI fhs r test_ValidHTTPz!TestDataSourceOpen.test_ValidHTTPb s5
W\\-//
*
*
r c t }t t | j j | | j | d S # t $ r!}t |j d u Y d }~d S d }~ww xY wr )r5 r OSErrorrG rQ r errno)rI r es r test_InvalidHTTPz#TestDataSourceOpen.test_InvalidHTTPg s gtw|S111 % GLL % % % AGtO $ $ $ $ $ $ $ $ $ %s A
A7A22A7c ^ t t | j j t d S r )r r rG _cacher5 rH s r test_InvalidHTTPCacheURLErrorz0TestDataSourceOpen.test_InvalidHTTPCacheURLErrorp s$ h0A0ABBBBBr c t | j }| j | }t | | d S r )r+ rD rG rQ r r& )rI
local_filerR s r test_ValidFilez!TestDataSourceOpen.test_ValidFiles s@ #DK00
W\\*
%
%
r c n t | j }t t | j j | d S r )r. rD r rU rG rQ )rI invalid_files r test_InvalidFilez#TestDataSourceOpen.test_InvalidFiley s- ' 44 gtw|\:::::r c dd l }n## t $ r t j Y nw xY wt j | j d }| |d }| t | | j | }|
}| t t | d S )Nr z
foobar.txt.gzw)gzip ImportErrorpytestskipr% r* joinrD rQ write
magic_liner& rG readliner )rI rd filepathfpresults r test_ValidGzipFilez%TestDataSourceOpen.test_ValidGzipFile} s KKKK KMMMMM 7<< _==
YYx
%
%
W\\(
#
#
Z((((( ''c dd l }n## t $ r t j Y nw xY wt j | j d }| |d }| t | | j
| }| }| t t | d S )Nr zfoobar.txt.bz2rc )bz2re rf rg r% r* rh rD BZ2Fileri rj r& rG rQ rk r )rI rr rl rm rn s r test_ValidBz2Filez$TestDataSourceOpen.test_ValidBz2File s JJJJ KMMMMM 7<< -=>>
[[3
'
'
W\\(
#
#
Z(((((rp N) __name__
__module__ __qualname__rJ rN rS rX r[ r^ ra ro rt r r r rA rA Y s 5 5 5
% % %C C C ; ; ;) ) ) ) ) ) ) )r rA c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestDataSourceExistsc h t | _ t j | j | _ d S r rC rH s r rJ z!TestDataSourceExists.setup_method rK r c 2 t | j | `d S r rM rH s r rN z$TestDataSourceExists.teardown_method rO r c l t | j t d S r )r rG existsr rH s r rS z#TestDataSourceExists.test_ValidHTTP s( }//00000r c n t | j t d d S NF)r rG r} r5 rH s r rX z%TestDataSourceExists.test_InvalidHTTP s, TW^^O$5$566>>>>>r c " t | j }t | j | t }t | }t | j | t
| d S r )r+ rD r rG r} r r )rI r localdirs r r^ z#TestDataSourceExists.test_ValidFile ss --w''(((99 **w''(((xr c ~ t | j }t | j | d d S r )r. rD r rG r} rI r s r ra z%TestDataSourceExists.test_InvalidFile s5 "4;//TW^^G,,e44444r N) ru rv rw rJ rN rS rX r^ ra r r r ry ry sn 5 5 5 1 1 1? ? ? 5 5 5 5 5r ry c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestDataSourceAbspathc t j t | _ t j | j | _ d S r )r% r* abspathr rD rE rF rG rH s r rJ z"TestDataSourceAbspath.setup_method s3 googii00 ' 44r c 2 t | j | `d S r rM rH s r rN z%TestDataSourceAbspath.teardown_method rO r c ^ t t \ }}}}}}t j | j || t j d }t || j
t d S N/) r
r r% r* rh rD stripsepr rG r )rI schemenetlocupathpmsqryfrg
local_paths r rS z$TestDataSourceAbspath.test_ValidHTTP s /7
/H/H,sCW\\$+v"'++bf"5"5";";C"@"@B B
Z !A!ABBBBBr c t | j }t j | d }t || j | t || j | d S N)r+ rD r% r* splitr rG r )rI r tmpfilenames r r^ z$TestDataSourceAbspath.test_ValidFile sk --gmmG,,R0 Wdgook : :;;;Wdgoog 6 677777r c d t t \ }}}}}}t j | j || t j d }t || j
t k d S r ) r
r5 r% r* rh rD r r r rG r r )rI r r r r r r invalidhttps r rX z&TestDataSourceAbspath.test_InvalidHTTP s /78I8I/J/J,sCgll4;#(;;rv#6#6#<#z7TestDataSourceAbspath.test_sandboxing.. s# RW__TW__Q-?-? @ @ r ) r+ rD r% r* r r r
startswithr5 malicious_filesr0 )rI r r tmp_pathfns ` r test_sandboxingz%TestDataSourceAbspath.test_sandboxing sP --gmmG,,R0 @@@@ ))44T[AABBB **++66t{CCDDD!!,,T[99:::%%00==>>>! : :B HHYr\**55dkBB C C C HHRLL++DK88 9 9 9 9 : :r c : t j } dt _ | | | | | |t _ d S # |t _ w xY wN\)r% r rS r^ rX ra r rI orig_os_seps r test_windows_os_sepz)TestDataSourceAbspath.test_windows_os_sep s f !BF ! ! ! ! ! ! ! ! # # # ! ! # # # " " " BFFF[BF s A0B BN) ru rv rw rJ rN rS r^ rX ra r r r r r r r s 5 5 5 C C C 8 8 8A A A 9 9 9 : : :
!
!
!
!
!r r c , e Zd Zd Zd Zd Zd Zd ZdS )TestRepositoryAbspathc t j t | _ t j t | j | _ d S r ) r% r* r r rD rE
Repositoryr7 reposrH s r rJ z"TestRepositoryAbspath.setup_method s9 googii00 *=??DKHH
r c 2 t | j | `d S r r rD r rH s r rN z%TestRepositoryAbspath.teardown_method t{ JJJr c l t t \ }}}}}}t j | j j || t j d }| j t }t || d S r ) r
r r% r* rh r _destpathr r r r= r ) rI r r r r r r r rl s r rS z$TestRepositoryAbspath.test_ValidHTTP s /7
/H/H,sCW\\$*"6"'++bf"5"5";";C"@"@B B
:%%n&6&677Z*****r c n fd}t |t j t D ]j}t |t
|z j t || j kd S )Nc r t j j | S r )r% r* r r r s r r z7TestRepositoryAbspath.test_sandboxing.. s% RW__TZ-?-?-B-B C C r )r r= r rD r r0 )rI r r s ` r r z%TestRepositoryAbspath.test_sandboxing
s CCCC ))**55dkBBCCC! : :B HHYr\**55dkBB C C C HHRLL++DK88 9 9 9 9 : :r c t j } dt _ | | |t _ d S # |t _ w xY wr )r% r rS r r s r r z)TestRepositoryAbspath.test_windows_os_sep sW f !BF ! ! ! " " " BFFF[BF s 4A A N)ru rv rw rJ rN rS r r r r r r r sb I I I + + +: : :! ! ! ! !r r c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestRepositoryExistsc t | _ t j t | j | _ d S r )r rD rE r r7 r rH s r rJ z!TestRepositoryExists.setup_method s+ ii *=??DKHH
r c 2 t | j | `d S r r rH s r rN z$TestRepositoryExists.teardown_method# r r c | t | j }t | j | d S r )r+ rD r r r} r s r r^ z#TestRepositoryExists.test_ValidFile' s5 --
!!'**+++++r c ~ t | j }t | j | d d S r )r. rD r r r} r s r ra z%TestRepositoryExists.test_InvalidFile, s7 "4;//TZ&&w//77777r c l t | j t d S r )r r r} r rH s r test_RemoveHTTPFilez(TestRepositoryExists.test_RemoveHTTPFile0 s*
!!-//2233333r c 8 t }t | \ }}}}}}t j | j j | }t j |d t | } t | j
| d S )Ni ) r r
r% r* rh r r mkdirr+ r r} )
rI localfiler r r r r r r r s
r test_CachedHTTPFilez(TestRepositoryExists.test_CachedHTTPFile3 s !OO 08 /B/B,sCW\\$*"6??
V$$$ ,,
!!'**+++++r N) ru rv rw rJ rN r^ ra r r r r r r r sq I I I , , ,
8 8 84 4 4 , , , , ,r r c e Zd Zd Zd Zd ZdS ) TestOpenFuncc , t | _ d S r )r rD rH s r rJ zTestOpenFunc.setup_method@ s ii r c . t | j d S r )r rD rH s r rN z TestOpenFunc.teardown_methodC s t{r c t | j }t j || j }t | | t j | }t | | d S )N)destpath)r+ rD rE rQ r r& )rI r] rm s r test_DataSourceOpenz TestOpenFunc.test_DataSourceOpenF sj #DK00
_Z$+
>
>
>
_Z
(
(
r N)ru rv rw rJ rN r r r r r r ? sA r r c X t j } | `| d S r )rE rF
_istmpdest__del__)rG s r test_del_attr_handlingr Q s,
B
JJLLLLLr r )0r% rf tempfiler r r shutilr numpy.lib._datasourcelib _datasourcerE
numpy.testingr r r urllib.requestrequestr urllib.parser
urllib.errorr r r r r r0 r1 r3 r4 r rj r+ r. r r5 r7 r: r= r? rA ry r r r r r r r r r s
9 9 9 9 9 9 9 9 9 9 * * * * * * * * * > > > > > > > > > > ' ' ' ' ' ' ! ! ! ! ! ! ! ! ! ! ! !4 4 4 4 * * *) ) )
% +
@ @ @ *
' ' ' B) B) B) B) B) B) B) B)J5 5 5 5 5 5 5 5<>! >! >! >! >! >! >! >!B ! ! ! ! ! ! ! !B , , , , , , , ,B $
r