birUddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m ZddlmZddgZej$dZej(j*efZd ed <e ej2e d kDr,ej$d Zeej6j8fzZej:d dZdZdZ dZ!ejDjGdddgdZ$dZ%dZ&dZ'dZ(ejDjRejDjGddgdZ*ejDjWddd ejDjRejDjYd!"d#Z-ejDjRd$Z.y)%) annotationsN) open_files)Version)tmpdirabrequestsztuple[type[Exception], ...]errsz0.7.4aiohttpmodule)scopec#dKt5}tD]I}ttjj ||d5}|j ddzdddKtjdddg}tj||}d} tjd  ||j%dddy#1swYxYw#tjj$r5}tj d |d z}|d kr t#d |Yd}~nd}~wwxYw#1swYyxYww) Nwbai'z-mz http.server8999)cwd zhttp://localhost:8999g?rzServer did not appear)rfilesopenospathjoinwritesys executable subprocessPopenr get exceptionsConnectionErrortimesleep RuntimeError terminate)dfnfcmdptimeoutes U/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/bytes/tests/test_http.py dir_serverr-s  Q &Bbggll1b)40 &Au % & & &~~t]F;   Sa ( G 45  % & &&&66 G 33Q;&'>?QF G s^ D04D$C8D$C$D$9 D0C D$D ++DD$D  D$$D-)D0c4d}td}t||zd}|5}|j}dddttj j ||d5}|jk(sJ dddy#1swYSxYw#1swYyxYw)Nhttp://localhost:8999/rrbrrreadrrrr)r-rootr&r'dataexpecteds r, test_simpler63s #D qB4"9a A avvx bggll:r*D 1'Xx}}&&&''''sB"BB Bcd}td}t||zd}ttjj ||d5}|j }|5}|j d}||ddk(sJ|jdk(sJ|jd|j d}||ddk(sJ|jdd|jdk(sJ ddddddy#1swYxYw#1swYyxYw)Nr/rr0) rrrrrrr2locseek)r-r3r&r'r5r4s r,test_locr>=s #D qB4"9a A bggll:r*D 1 X==?  !66!9D8BQ<' ''55A: : FF1I66!9D8BQ<' '' FF1aL55A: :       s%C4A :EE  E EE"%E. *E11E:=F F  Fcvd}ttDcgc]}||z c}}t|tD]g\}}|5}ttj j ||d5}|j|jk(sJ ddddddiycc}w#1swYxYw#1swYxYw)Nr/r0)rrziprrrrr2)r-r3r'fsf2r5s r, test_filesr[s #D u-!TAX- .BR32  3!bggll:r2D9 3Xvvx8==?222 3 3 33. 3 3 3 3s( B+B/&$B# B/#B, (B//B8 ctd}t|dz}|djdk(sJ|djdk(sJy)Nr/*rzhttp://localhost:8999/ar:zhttp://localhost:8999/b)rr)r-r3rYs r,test_open_globr^sC #D D3J B a5::2 22 2 a5::2 22 2enginepyarrowcntjddtjd}tj||jdg|j}|jj t tdk(sJ|jj gdk(sJy) Nr z2.21.0) minversiondask.dataframezthttps://github.com/Parquet/parquet-compatibility/raw/master/parquet-testdata/impala/1.1.1-NONE/nation.impala.parquet)r`) n_nationkeyn_name n_regionkey n_comment) rN importorskip read_parquetcomputerftolistlistrangecolumns)r`dddfs r, test_parquetrss   x8   - .B   $   gi >> "d59o 55 5 ::   "W WW Wr_rz(https://github.com/dask/dask/issues/3696)reruns reruns_delayreasonz Flaky test.)rwcxddg}tj|}|jdk(sJ|jy)Nrhttps://raw.githubusercontent.com/weierophinney/pastebin/master/public/js-src/dojox/data/tests/stores/patterns.csvzhttps://en.wikipedia.orgr8)db read_text npartitionsrl)urlsrs r,test_bagr~s= D" D TA ==A  IIKr_cttjd}d}|j|}|jy)Nrdry)rNrjread_csvrl)rqurlrs r, test_read_csvrs7   - .B D CAIIKr_)/ __future__rrrrr!rLrN fsspec.corerpackaging.versionrdask.bagbagrz dask.utilsrrrjr rRequestExceptionrTr __annotations__rMr client_exceptionsClientResponseErrorfixturer-r6r>rDmark parametrizerHrQrVr[r^networkrsflakyskipr~rr_r,rs"  "% c  6  z * ((%! 6  !11!f!!),G 7,,@@B BDh ,'" 'e}5 .6 . 34033 K X  X  A&P' ( r_