
    >ie                       U d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlm	Z	 d dl
mZ d dlmZ d dlmZ ddgZ ej        d          Zej        j        efZd	ed
<    eej                   ed          k    r  ej        d          Zeej        j        fz   Z ej        d          d             Zd Z d Z!d Z"ej#        $                    dddg          d             Z%d Z&d Z'd Z(d Z)ej#        j*        ej#        $                    dddg          d                         Z+ej#        ,                    ddd !          ej#        j*        d"                         Z-ej#        j*        d#             Z.dS )$    )annotationsN)
open_files)parse)tmpdirabrequestsztuple[type[Exception], ...]errsz0.7.4aiohttpmodule)scopec               #    K   t                      5 } t          D ]`}t          t          j                            | |          d          5 }|                    ddz             d d d            n# 1 swxY w Y   at          j        dddg}t          j
        ||           }d}	 	 t                              d
           nQ# t          j        j        $ r9}t          j        d           |dz  }|dk     rt#          d          |Y d }~nd }~ww xY wm| V  |                                 d d d            d S # 1 swxY w Y   d S )Nwb   ai'  z-mzhttp.server8999)cwd
   Tzhttp://localhost:8999g?r   zServer did not appear)r   filesopenospathjoinwritesys
executable
subprocessPopenr	   get
exceptionsConnectionErrortimesleepRuntimeError	terminate)dfnfcmdptimeoutes          :lib/python3.11/site-packages/dask/bytes/tests/test_http.py
dir_serverr-      s     	 Q 	& 	&Bbgll1b))400 &Au%%%& & & & & & & & & & & & & & & ~t]F;Sa(((	GG4555&6 G G G
33Q;;&'>??QF ;;;;G		G 	%                 s_   8D5	A."D5.A22D55A26-D5$B?>D5?D/DD5DD55D9<D9c                t   d}t           d         }t          ||z             d         }|5 }|                                }d d d            n# 1 swxY w Y   t          t          j                            | |          d          5 }||                                k    sJ 	 d d d            d S # 1 swxY w Y   d S )Nhttp://localhost:8999/r   rbr   r   readr   r   r   r   )r-   rootr&   r'   dataexpecteds         r,   test_simpler6   3   s(   #D	qB4"9a A	
 avvxx              	bgll:r**D	1	1 'Xx}}&&&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's#   AAAB--B14B1c                ^   d}t           d         }t          ||z             d         }t          t          j                            | |          d          5 }|                                }|5 }|                    d          }||d d         k    sJ |j        dk    sJ |                    d           |                    d          }||d d         k    sJ |                    dd           |j        dk    sJ 	 d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nr/   r   r0               )	r   r   r   r   r   r   r2   locseek)r-   r3   r&   r'   r5   r4   s         r,   test_locr>   =   s   #D	qB4"9a A	bgll:r**D	1	1 
X==?? 	!66!99D8BQB<''''5A::::FF1III66!99D8BQB<''''FF1aLLL5A:::::	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
s7   D"-BD
>D"
D	D"D	D""D&)D&c                   d}t           d         }ddi}t          ||z   |          d         }|5 }|                    d          |                    d          z   }d d d            n# 1 swxY w Y   t          t          j                            | |          d	          5 }||                                k    sJ 	 d d d            d S # 1 swxY w Y   d S )
Nr/   r   DatezWed, 21 Oct 2015 07:28:00 GMT)headersr:   )lengthr0   r1   )r-   r3   r&   rA   r'   r4   r5   s          r,   test_fetch_range_with_headersrD   N   sQ   #D	qB67G4"9g...q1A	
 4avvQv!&&&"3"334 4 4 4 4 4 4 4 4 4 4 4 4 4 4	bgll:r**D	1	1 'Xx}}&&&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's#   .A**A.1A.#CCC
block_sizei c                   d}t           d         }t          ||z             d         }t          t          j                            | |          d          5 }|                                }|5 }|                    d          |d d         k    sJ |                    d           |                    d          |d d         k    sJ |                    d          |dd         k    sJ |                    dd           |                                |dd          k    sJ 	 d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nr/   r   r0   r      ir8   )r   r   r   r   r   r   r2   r=   )r-   rE   r3   r&   r'   r5   s         r,   test_opsrH   Z   s   #D	qB4"9a A	bgll:r**D	1	1 	.X==?? 	.!66"::#2#....FF1III66"::#2#....66"::"R%0000FF3NNN6688x~-----	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	.	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	.s7   E	-B7D1%E	1D5	5E	8D5	9E		EEc                t   d}t           d         }t          ||z   d          d         }t          t          j                            | |          d          5 }|                                }|5 }|                                |k    sJ |j        t          |          k    sJ 	 d d d            n# 1 swxY w Y   t           d         }t          ||z   d          d         }|5 }t          t          j                  t          d          k     rVt          j        t                    5  |                    d          |d d         k    sJ 	 d d d            n# 1 swxY w Y   n#|                    d          |d d         k    sJ d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )	Nr/   r   r8   )rE   r0   r:   z	2021.11.1r   )r   r   r   r   r   r   r2   sizelenparse_versionfsspec__version__pytestraises
ValueError)r-   r3   r&   r'   r5   s        r,   test_ops_blocksizerR   k   s   #D	qB4"9+++A.A	bgll:r**D	1	1 3X==?? 	+!6688x''''6S]]*****		+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 1XtbyQ///2 	3!V/00=3M3MMM]:.. 7 766"::#2#666667 7 7 7 7 7 7 7 7 7 7 7 7 7 7 vvbzzXcrc]2222	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	33 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s   F-/5B1%F-1B5	5F-8B5	9,F-%AF)$EFEF!E"'F	F-F	F-F	F--F14F1c                   t          d          d         }t          j        t                    5  |5 }|                                 d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   t          d          d         }t
          }t          j        |          5  |5 }|                                 d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   d}t          d         }t          ||z   d          d         }t          j        t                    5  |5  	 d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   t          ||z             d         }|5 }t          j        t                    5  |	                    d           d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nz"http://localhost:8999/doesnotexistr   zhttp://nohost/r/   r   )moderC   )
r   rO   rP   r
   r2   FileNotFoundErrorr   NotImplementedErrorrQ   r=   )r-   r'   r5   r3   r&   s        r,   test_errorsrW      s   788;A	t		   	!FFHHH	 	 	 	 	 	 	 	 	 	 	 	 	 	 	               	#$$Q'A H	x	 	    	!FFHHH	 	 	 	 	 	 	 	 	 	 	 	 	 	 	               $D	qB4"94(((+A	*	+	+   	 		 	 	 	 	 	 	 	 	 	 	 	 	 	 	               	4"9a A	
 a]:&& 	 	FF2JJJ	 	 	 	 	 	 	 	 	 	 	 	 	 	 	                 s   A*AA*A	A*A	A**A.1A.%C (C	=C 	C	C C	C  C$'C$-E0D>2E>E	EE	EEE:GF6*G6F:	:G=F:	>GGGc                   dt          fdt          D                       }t          |t                    D ]\  }}|5 }t          t          j                            | |          d          5 }|                                |                                k    sJ 	 d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   d S )Nr/   c                    g | ]}|z   S  rZ   ).0r'   r3   s     r,   
<listcomp>ztest_files.<locals>.<listcomp>   s    ---!TAX---    r0   )r   r   zipr   r   r   r   r2   )r-   fsr'   f2r5   r3   s        @r,   
test_filesra      s>   #D	----u---	.	.BR 3 32 	3!bgll:r22D99 3Xvvxx8==??222223 3 3 3 3 3 3 3 3 3 3 3 3 3 3	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	33 3s6    /C /-B)C )B--C 0B-1C  C	C	c                z    d}t          |dz             }|d         j        dk    sJ |d         j        dk    sJ d S )Nr/   *r   zhttp://localhost:8999/ar:   zhttp://localhost:8999/b)r   r   )r-   r3   r_   s      r,   test_open_globrd      sN    #D	D3J		Ba5:22222a5:2222222r]   enginepyarrowfastparquetc                   t          j        dd           t          j        d          }t          j        |            |                    dg|                                           }|j                                        t          t          d                    k    sJ |j                                        g dk    sJ d S )	Nr	   z2.21.0)
minversiondask.dataframezthttps://github.com/Parquet/parquet-compatibility/raw/master/parquet-testdata/impala/1.1.1-NONE/nation.impala.parquet)re      )n_nationkeyn_namen_regionkey	n_comment)	rO   importorskipread_parquetcomputerl   tolistlistrangecolumns)re   dddfs      r,   test_parquetry      s     
x8888		-	.	.B
	$	

  
 
 
 gii  >  ""d599oo5555:"W"W"WWWWWWWr]   r      z(https://github.com/dask/dask/issues/3696)rerunsreruns_delayreasonc                 x    ddg} t          j        |           }|j        dk    sJ |                                 d S )Nrhttps://raw.githubusercontent.com/weierophinney/pastebin/master/public/js-src/dojox/data/tests/stores/patterns.csvzhttps://en.wikipedia.orgr8   )db	read_textnpartitionsrr   )urlsr   s     r,   test_bagr      sG    	D"D
 	TA=AIIKKKKKr]   c                     t          j        d          } d}|                     |          }|                                 d S )Nrj   r   )rO   rp   read_csvrr   )rw   urlr   s      r,   test_read_csvr      sB    		-	.	.B	D  	CAIIKKKKKr]   )/
__future__r   r   r   r   r!   rM   rO   fsspec.corer   packaging.versionr   rL   dask.bagbagr   
dask.utilsr   r   rp   r	   r   RequestExceptionrU   r
   __annotations__rN   r   client_exceptionsClientResponseErrorfixturer-   r6   r>   rD   markparametrizerH   rR   rW   ra   rd   networkry   flakyr   r   rZ   r]   r,   <module>r      s   " " " " " " " 				     



    " " " " " " 4 4 4 4 4 4            	c
6z** (%     =#$$}}W'='===!f!),,G7,@BBD h   ,' ' '  "	' 	' 	' e}55. . 65. 3 3 34  03 3 33 3 3 I}#=>>X X ?> X  A&P    	 	  	     r]   