>ieddlmZddlZddlZddlmZejj Z ddl m Z ddlmZmZejdZeejZdZdZdZd Zd Zejeed kd dZejeed kd dZdS)) annotationsN)parse) assert_eq same_keyscupyctdd}tj|dd}tj|ddd}|jd ksJtj gd gd gd gd gdgdgd gd gdgdgdgdg }t||dttj|ddd|sJdS)N@r r Fchunksasarrayrr))rr) rrrr rr rr) r rr ) r#r$) r+r,) !"#$r3r4%&') ()*+,r;r<-./) 01234rCrD567) 89:;<rKrL=>? check_type) rarangereshapeda from_arrayoverlapoverlap_internalrnparrayrr)xdgexpecteds Blib/python3.11/site-packages/dask/array/tests/test_cupy_overlap.pytest_overlap_internalr_s3 B''A a666A ##A1||44A 8' ' ' ' 'x * * * 2 2 2 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 H"ae,,,, RZ00aLLAA1 E EEEEEEctd}tj|dd}tj|dddd }|jd ksJdS) N)r8rL)rrFrrrrreflect)axesboundary))r r r r )rrrrrr)ronesrTrUrV trim_internalrrZr[es r^test_trim_internalri,se (A a%888A   Q1  JJA 89 9 9 9 9 9 9r`ctdd}tj|dd}tj|dd}|jd|jdd zksJ|jd |jd ksJt|d ddf|d ddft|dddf|d ddfdS) Nr r r Frrraxisdepthr r) rrRrSrTrUrVperiodicshaperrgs r^ test_periodicrr4s B''A a666A AAQ//A 71:a ' ' ' ' 71: # # # # a111gqQQQx    a111gqQQQx     r`ctd}tj|dd}tj|dd}t jgd}t||d tj|dd }t jgd }t||d dS) NrrFrrrrk)rrrrrrr rrrr rrr rPr rrrrrr rrrr rr) rrRrTrUrVrbrXrYrrZr[rhr]s r^ test_reflectrv@ BA a666A 11A..AxBBBCCH ae,,,, 11A..Ax<<<==H ae,,,,,,r`ctd}tj|dd}tj|dd}t jgd}t||d tj|dd }t jgd }t||d dS) NrrFrrrrk)rrrrrrr rrrr rrrrPrrt) rrRrTrUrVnearestrXrYrrus r^ test_nearestrzMrwr`z6.4.0zBRequires CuPy 6.4.0+ (with https://github.com/cupy/cupy/pull/2418))reasonc2tdd}tj|dd}tj|ddd }|jd|jdd zksJ|jd |jd ksJt|d ddftj d |j dzdt|dddftj d |j dzddS)Nr r r Frrrr)rlrmvaluer rr )dtyperPrn) rrRrSrTrUrVconstantrqrrXrer~rgs r^ test_constantrZs B''A a666A AAQb99A 71:a ' ' ' ' 71: # # # # a111grwq00025%HHHH aAAAh111B65IIIIIIr`cptdd}tj|dd}tj|dddd d d}tjgd gd gd gd gdgdgdgdgdgdgd gd g }t||ddS)Nr r r Frrrrrrp) rrrrrrrrrr) rrrrrr rrrr) rr rrrrrrrr ) r'r r!r"r#r$r%r&r'r ) r/r(r)r*r+r,r-r.r/r() r7r0r1r2r3r4r5r6r7r0) r?r8r9r:r;r<r=r>r?r8) rGr@rArBrCrDrErFrGr@) rOrHrIrJrKrLrMrNrOrHrP) rrRrSrTrUrV boundariesrXrYrrus r^test_boundariesrjs B''A a666A aQ11/D/DEEAx * * * * * * * * * 1 1 1 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 * * * * * * H ae,,,,,,r`) __future__rnumpyrXpytestpackaging.versionr parse_versionmarkgpu pytestmark dask.arrayrYrTdask.array.utilsrr importorskipr __version__ cupy_versionr_rirrrvrzskipifrrr`r^rs"""""" 444444 [_ 11111111v6""}T-.. FFF8::: ! ! ! - - - - - -==))) O J J  J==))) O-- ---r`