o 3ήc"@s|dZddlZedZedddlZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZdS)z?Test trophic levels, trophic differences and trophic coherence NnumpyscipycCsBt}|dd|ddt|}|ddddksJdS) zTrivial exampleabc)rrrN)nxDiGraphadd_edgetrophic_levels)GdrX/tmp/pip-target-vg8gfxp4/lib/python/networkx/algorithms/centrality/tests/test_trophic.pytest_trophic_levels s    rcCst}|jdddd|jdddd|jdddd|jdddd|jdd d d|jddd d|jdd d d|jd ddd|}d d|jD}|D]}||qRtjj|j }t gdgdgdgdg}t | |s}Jt|jdksJ|jd|jdksJ|jdt|ksJ|jd}t |}t j||}t |t |} t gd} t | | sJdddddd} t|} | D]\}} | |}|tj| ddksJqdS)zSExample from Figure 5 in Stephen Levine (1980) J. theor. Biol. 83, 195-207 rrg?)weightr 皙?g?333333?333333?ffffff?cSsg|] \}}|dkr|qS)rr).0nidrrrr 'sz.test_trophic_levels_levine..)rrgr)rrrr)rrrr)rrrrr)rgT@gq?g|kʚ@gT@gX[⎸@g|kʚ @)rrr rrHz>absN)r r r copy in_degree remove_nodelinalg graphmatrixadjacency_matrixTnparray array_equaltodenselenshapeeyeinvasarrayonesallcloser itemspytestapprox)SS2zrq expected_qnniny expected_y expected_drlevelexpected_levelrrrtest_trophic_levels_levinesH    rBcCshtddgddgg}tj|tjd}t|}|dtjdddks%J|dtjdddks2JdS)Nrr create_usingrrr)r'r(r from_numpy_arrayr r r3r4)matrix_arrrrrtest_trophic_levels_simpleOs  rGcCstgdgdgdgdg}tj|tjd}t|}gd}tdD]}||tj||dd ks7Jq&tgd gd gdgdg}tj|tjd}t|}gd }t d |t d|tdD]}||tj||dd ksyJqhdS)N)rrrr)rrrrrrrrrrrrrC)rrr rrrrrrrrrrrr)rrg@g @zCalculated result: zExpected Result: ) r'r(r rEr r ranger3r4print)matrixrrexpected_resultindrrr test_trophic_levels_more_complexWs2        rQcCstgdgdgdgdgdg}tgdgdgdgd gd g}ttj|td }tj|tjd }t |}t d D]}||t j ||d dksXJqGdS)Nrrrrrrrrrrrrrrrrrrrrrrrrr)rrrr)r?rr)rrrrr)rrYrrrY)rrrrrrrCrr) r'r(ravelr#r.r0r rEr r rLr3r4)rNKresult_1rresult_2rPrrr%test_trophic_levels_even_more_complexys(   r^cCsjtd}tj|tjd}ttj }t|Wdn1s#wYd}|t |j vs3JdS)z;Should raise an error with graphs with only non-basal nodesrrCNTrophic levels are only defined for graphs where every node has a path from a basal node (basal nodes are nodes with no incoming edges).) r'identityr rEr r3raises NetworkXErrorr strvalue)rNremsgrrr#test_trophic_levels_singular_matrixs  rgcCs t}|dd|dd|dd|dd|ddttj }t|Wdn1s8wYd}|t|jvsHJt}|dd|dd|ddttj }t|Wdn1stwYd}|t|jvsJdS)zShould fail to compute if there are any parts of the graph which are not reachable from any basal node (with in-degree zero). rrrrNr_) r r r r3rarbr rcrd)rrerfrrr'test_trophic_levels_singular_with_basals,          rhcCs tddgddgg}tj|tjd}t|}|dtjdddks%Jtgdgdgd gd g}tj|tjd}t|}|dtjdddksPJ|d tjd ddks]J|d tjdddksjJ|dtjdddkswJ|dtjdddksJdS)NrrrC)rrrrrJrKrHrI)rr?)rrrX)rr ?)rr ?)r'r(r rEr trophic_differencesr3r4)rFrdiffsmatrix_brrrtest_trophic_differencess"  rocCsntddgddgg}tj|tjd}tj|dd}|tjdddks%Jtgdgd gd gd g}tj|tjd}tj|dd}|tjtgd ddksUJtgdgd gd gd g}tj|tjd}tj|dd}|tjtgd ddksJtgdgd gd gd g}tj|tjd}tj|dd}|tjtgd ddksJdS)NrrrCF cannibalismrrrJrKrHrIrrirXrkrj)rrrr r'r(r rEr trophic_incoherence_parameterr3r4std)rFrr8rnmatrix_cmatrix_drrr1test_trophic_incoherence_parameter_no_cannibalisms>  $rxcCs tddgddgg}tj|tjd}tj|dd}|tjdddks%Jtgdgd gd gd gd g}tj|tjd}tj|dd}|tjd ddksSJtgdgdgdgdg}tj|tjd}tj|dd}|tjtgdddksJdS)NrrrCTrprrrRrSrTrUrVrrJrKrHrIrrrs)rFrr8rnrvrrr.test_trophic_incoherence_parameter_cannibalisms.$ry)__doc__r3 importorskipr'networkxr rrBrGrQr^rgrhrorxryrrrrs   :"( .