a d@sddlZddlmmZddlmZmZmZddl m Z ej dej dejdZdedddf<dedddf<dZd d Zd d Zd dZddZddZedddgdddZddZdS)N)assert_array_equalassert_almost_equal parametrize)expected_warnings)r)Zdtypez|\A\Zc Cstdtgtjtdd}Wdn1s20Y|dg\}}|d}t|gdgdgdgd gd gd gd gd gt|gddS)NUpgrading NumPyTZfully_connectedr)?rrrrrrrrrrrrrr)rrrrrrrr)rrr@rrrr)rrrr@rrr)rrrrr@rr)rrrrrr@r)rrrrrrr@ r rrrrrrrrrrrrrr rr rwarning_optionalmcpMCPa find_costs tracebackrmcostsr+ return_pathr0e/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/skimage/graph/tests/test_mcp.py test_basics$,  r2cCsttdktjd}gd}ttdkdd}ttdktjd}tdtgtj|dd}Wdn1st0Y|dg\}}| d }t ||t ||tdtgtj|dd}Wdn1s0Y|dg\}}| d }t ||t ||dS) Nrr) r rrrrr r!r"r#r$rr Tr r r$) npwherer)infrr&r'r(r*r+r)Zexpected_costsZ expected_pathZtest_negZtest_infr-r.r+r/r0r0r1 test_neg_inf.s  ,   ,  r6cCsjtdtg&tjtdddd\}}Wdn1s:0Yt|tddt|gddS)Nr r r T)Z geometricrr) rr&r'Zroute_through_arrayr)rr3sqrtr)r/Zcostr0r0r1 test_routeJs (r8c Cstdtgtjtdd}Wdn1s20Y|dg\}}|d}t|gdgdgdgd gd gd gd gd gt|gddS)Nr Fr r r )rrrrrrrrr)rrrrrrrr)rrrrrrrr)rrrrrrrr)rrrrrrrr)rrrrrrrr)rrrrrrrg@) r rrrr)rrr r!r"r#r$)rrr r%r,r0r0r1test_no_diagonal]s$,  r9c CsddtdDddtddD}tdtgtjt|d}Wdn1sX0Y|dg\}}t|gd gd gd gd gd gd gd gd gt|d sJt||j dS)NcSsg|] }d|fqSrr0.0ir0r0r1 |z test_offsets.. cSsg|]}d| fqSr:r0r;r0r0r1r>|r?rr )offsetsr )rBrBrBrBrBrBrB)rBrBrBrBrBrBrrB) r@rr)r@rrrrrrr rA) rangerr&r'r(r)r*rhasattrrA)rAr-r.r+r0r0r1 test_offsets{s"&, rJshape)drL)rrrErcCs t|dS)N) _test_randomrKr0r0r1 test_crashingsrPc stjjtj}dgtdgttjttg}fddtdD}tdt gt j |dd}Wdn1s0Y| |\}}fd dtdDD]}| |q|| |||D]}| |q|||fS) Nrrcs&g|]}tjttqSr0r3randomrandlenastypeintr;rOr0r1r>sz _test_random..rr Tr cs&g|]}tjttqSr0rQr;rOr0r1r>s)r3rRrSrUfloat32rTrVrHrr&r'r(r*r+Z_reset) rKr)ZstartsZendsr-r.rAZpointendr0rOr1rNs& ,     rN)numpyr3Zskimage.graph.mcpgraphr'Zskimage._shared.testingrrrZskimage._shared._warningsrrRseedZonesrWr)r&r2r6r8r9rJrPrNr0r0r0r1s