o 3╬оcg<у@sFddlZddlZddДZGddДdГZGddДdГZGdd Дd ГZdS) щNcsJt|tГsJВ|d|ksJВ|d|ksJВ|dИ|dИf}|dИ|dИf}t|Гt|d|dГt|d|dГdksKJВЗfddД|DГ}|D]"}d|dkre|kshJВJВd|dkruИksxJВJВqVt|ddЕ|ddЕГD]\}}t|d|dГt|d|dГfdvsвJВqЖdS)Nrщ щcs$g|]}|dИ|dИfСqS)rй)┌.0┌uй┌cr·\/tmp/pip-target-vg8gfxp4/lib/python/networkx/algorithms/shortest_paths/tests/test_generic.py┌ s$z&validate_grid_path..))rr)rr)┌ isinstance┌list┌len┌abs┌zip)┌rr┌s┌t┌pr┌vrrr ┌validate_grid_paths8 ""2 rc@sАeZdZeddДГZddДZddДZddДZd d ДZddДZ d dДZ ddДZddДZddДZ ddДZddДZddДZddДZdS)┌TestGenericPathcCsЪddlm}|tаddбdddН|_tаdб|_tjdtабdН|_tаб|_ |j j dddd Н|j j dd dd Н|j j dddd Н|j j d ddd НdS) Nr)┌convert_node_labels_to_integersщr┌sorted)┌first_label┌orderingщ)┌create_usingй┌weightщщщ■ )┌networkxr┌nx┌ grid_2d_graph┌grid┌cycle_graph┌cycle┌DiGraph┌directed_cycle┌neg_weights┌add_edge)┌cls┌cnltirrr ┌setup_classs zTestGenericPath.setup_classcCsЖtа|jddбgdвksJВtа|jddбgdвksJВtddddtа|jddбГtа|jddбgdвks8JВtj|jdddd НgdвksHJВtj|jdddd НgdвksXJВtddddtj|jdddd НГtj|jdddd НgdвksxJВtj|jdddd dНgdвksЙJВtj|jdddddНgdвksЪJВtj|jdddddНgd вksлJВtjt tj|jddНtаtj tj|jdбdS)Nrr!йrrr r!r)rщщrrщrr┌dijkstraйr┌method·bellman-ford)rr r!┌SPAMйr6щ)r$┌ shortest_pathr(rr&r*r+┌pytest┌raises┌ ValueError┌NodeNotFoundй┌selfrrr ┌test_shortest_path#s6 ■ ■ ■z"TestGenericPath.test_shortest_pathcCsдddgdgddgdЬ}tjtаdбddН}||ksJВtjtаdбdddН}||ks,JВtjtаdбddd d Н}||ks>JВtjtаdбdddd Н}||ksPJВdS)Nrrr йrrr r!й┌targetrйrErr4йrErr6r7)r$r;┌ path_graphйrA┌answer┌sprrr ┌test_shortest_path_targetEs z)TestGenericPath.test_shortest_path_targetcCstа|jddбdksJВtа|jddбdksJВtа|jddбdks$JВtj|jddddНdks2JВtj|jddddНdks@JВtj|jddddНdksNJВtj|jdddd d Нdks]JВtj|jddddd НdkslJВtjttj|jdd Нtаtjtj|jdбdS)Nrr!rr3r2rrrr4r5r7r8r9r:) r$┌shortest_path_lengthr(r&r*r<r=r>r?r@rrr ┌test_shortest_path_lengthVs( ¤ ¤ z)TestGenericPath.test_shortest_path_lengthcCsЮddddЬ}ttjtаdбddНГ}||ksJВtjtаdбdddН}||ks)JВtjtаdбdddd Н}||ks;JВtjtаdбddd d Н}||ksMJВdS)NrrrCr!rDrrFr4rGr7)┌dictr$rMrHrIrrr ┌ test_shortest_path_length_targetps z0TestGenericPath.test_shortest_path_length_targetcCs<tа|jdб}|dgdвksJВ|tа|jdбksJВtа|jdб}tdddd|dГtj|jdddН}|dgdвks@JВ|tа|jdбksKJВtj|jdddН}tdddd|dГtj|jdd dd Н}|dgdвksrJВ|tа|jdбks}JВtj|jdddd Н}|dgdвksСJВ|tа|jdбksЬJВdS)Nrr!r0rrr3rrr4)r6rr7)r$r;r(┌single_source_shortest_pathr&r┌single_source_dijkstra_pathйrArrrr ┌ test_single_source_shortest_pathБs z0TestGenericPath.test_single_source_shortest_pathc Cs|ttа|jdбГ}|ddddddddЬksJВ|ttа|jdбГks$JВttа|jdбГ}|ddks5JВttj|jddd НГ}|ddddddddЬksNJВ|ttа|jdбГks[JВttj|jddd НГ}|ddksnJВttj|jddd dНГ}|ddddddddЬksИJВ|ttа|jdбГksХJВttj|jddddНГ}|ddddddddЬksпJВ|ttа|jdбГks╝JВdSй Nrrr r!)rrr r!rr2r1щr1rrr4r5r7)rOr$rMr(┌"single_source_shortest_path_lengthr&┌"single_source_dijkstra_path_length┌&single_source_bellman_ford_path_lengthйrA┌ansrrr ┌'test_single_source_shortest_path_lengthХs, z7TestGenericPath.test_single_source_shortest_path_lengthcCsPtа|jб}|ddgdвksJВ|ttа|jбГksJВtа|jб}tdddd|ddГtj|jddН}|ddgdвksDJВ|ttа|jбГksPJВtj|jddН}tdddd|ddГtj|jdd d Н}|ddgdвksyJВ|ttа|jбГksЕJВtj|jddd Н}|ddgdвksЪJВ|ttа|jбГksжJВdS)Nrr!r0rrr3rrr4r5r7) r$r;r(rO┌all_pairs_shortest_pathr&r┌all_pairs_dijkstra_path┌all_pairs_bellman_ford_pathrSrrr ┌test_all_pairs_shortest_pathпs z,TestGenericPath.test_all_pairs_shortest_pathc CsАttа|jбГ}|dddddddddЬksJВ|ttа|jбГks$JВttа|jбГ}|dddks6JВttj|jdd НГ}|dddddddddЬksPJВ|ttа|jбГks\JВttj|jdd НГ}|dddkspJВttj|jdd dНГ}|dddddddddЬksЛJВ|ttа|jбГksЧJВttj|jdddНГ}|dddddddddЬks▓JВ|ttа|jбГks╛JВdSrU)rOr$rMr(┌all_pairs_shortest_path_lengthr&┌all_pairs_dijkstra_path_length┌"all_pairs_bellman_ford_path_lengthrZrrr ┌#test_all_pairs_shortest_path_length├s( z3TestGenericPath.test_all_pairs_shortest_path_lengthcCsRtаб}tа|tdГбtа|tddГбtа|ddбsJВtа|ddбr'JВdS)Nr!r2rr r)r$┌Graph┌add_path┌range┌has_pathйrA┌Grrr ┌ test_has_path█s zTestGenericPath.test_has_pathc CsLtаб}tа|gdвбtа|gdвбgdвgdвgttа|ddбГks'JВtаб}tа|gdвбtа|gdвбgdвgdвgttj|ddddНГksPJВtаб}tа|gdвбtа|gdвбgdвgdвgttj|dddddНГkszJВtаб}tа|gdвбtа|gdвбgdвgdвgttj|dddd dНГksдJВdS) Nr0)rщ щr!rr!rrr4r5r7)r$rerfr┌all_shortest_pathsrirrr ┌test_all_shortest_pathsтs,& z'TestGenericPath.test_all_shortest_pathscCsXtаtjбПtаdб}|аdбttа|ddбГWdГdS1s%wYdS)Nrr)r<r=r$┌NetworkXNoPathrH┌add_noderrnrirrr ┌test_all_shortest_paths_raise№s "¤z-TestGenericPath.test_all_shortest_paths_raisec CsRtаtбПtаdб}ttj|dddddНГWdГdS1s"wYdS)Nr rrrr8r5)r<r=r>r$rHrrnrirrr ┌test_bad_methods "■zTestGenericPath.test_bad_methodcCsЇtаб}tа|gdвбtа|gdвбd|jdd<ttj|dddddНГ}ttj|dddddНГ}ttj|dddd dНГ}ttj|dddd dНГ}t|Гtd dД|DГГksZJВt|ГtddД|DГГksiJВt|Гtd dД|DГГksxJВdS)N)rrr!r0r)rr rr!r4r5r7csєБ|] }|dddЕVqdSйNrrйrrrrr ┌ єАzKTestGenericPath.test_all_shortest_paths_zero_weight_edge..csrtrurrvrrr rwrxcsrtrurrvrrr rwrx)r$rerf┌edgesrrnr)rA┌g┌paths30d┌paths03d┌paths30b┌paths03brrr ┌(test_all_shortest_paths_zero_weight_edges& "z8TestGenericPath.test_all_shortest_paths_zero_weight_edgeN)┌__name__┌ __module__┌__qualname__┌classmethodr/rBrLrNrPrTr\r`rdrkrorrrsrrrrr rs "rc@sLeZdZddДZddДZddДZddДZd d ДZddДZd dДZ ddДZ dS)┌TestAverageShortestPathLengthcCє*tаtаdбб}|tjdddНksJВdS)Nrr чHп╝ЪЄ╫z>йr)r$┌average_shortest_path_lengthr'r<┌approxrZrrr ┌test_cycle_graphєz.TestAverageShortestPathLength.test_cycle_graphcCrЕ)Nr2r rЖrЗ)r$rИrHr<rЙrZrrr ┌test_path_graph"rЛz-TestAverageShortestPathLength.test_path_graphcCsДtаб}tj|tdГddНtj|ddН}|tjdddНks JВtаб}tj|tdГddНtj|ddН}|tjdddНks@JВdS) Nrr rrrrЖrЗr2йr$re┌ add_cyclergrИr<rЙrfйrArjr[rrr ┌ test_weighted&sz+TestAverageShortestPathLength.test_weightedcCs tаб}tj|tdГddНtj|dddН}|tjddd Нks!JВtj|dd dН}|tjddd Нks4JВtj|dddН}|tjddd НksGJВtаб}tj|tdГddНtj|dddН}|tjddd НkshJВtj|dd dН}|tjddd Нks{JВtj|dddН}|tjddd НksОJВdS) Nrr rrr4r5rrЖrЗr7zfloyd-warshallr2rНrПrrr ┌test_specified_methods0s( z4TestAverageShortestPathLength.test_specified_methodscCsRtаб}|аtdГб|аddбtаtjtj|б|а б}tаtjtj|бdS)Nr!rr) r$re┌add_nodes_fromrgr,r<r=┌ NetworkXErrorrИ┌to_directed)rArzrrr ┌test_disconnectedGsz/TestAverageShortestPathLength.test_disconnectedcCstаб}tа|бdks JВdS)z╔Tests that the trivial graph has average path length zero, since there is exactly one path of length zero in the trivial graph. For more information, see issue #1960. rN)r$┌ trivial_graphrИrirrr ┌test_trivial_graphOsz0TestAverageShortestPathLength.test_trivial_graphcCs@tаtjбПtаtаббWdГdS1swYdS)N)r<r=r$┌NetworkXPointlessConceptrИ┌ null_graphr@rrr ┌test_null_graphZs" z-TestAverageShortestPathLength.test_null_graphcCsJtаtбПtаdб}tj|dddНWdГdS1swYdS)Nr rr8r5)r<r=r>r$rHrИrirrr rs^s "■z-TestAverageShortestPathLength.test_bad_methodN)rАrБrВrКrМrРrСrХrЧrЪrsrrrr rДs rДc@s eZdZeddДГZddДZdS)┌"TestAverageShortestPathLengthNumpycCsddl}|аdбadS)Nr┌numpy)r<┌importorskip┌np)r-r<rrr r/esz.TestAverageShortestPathLengthNumpy.setup_classcCsxtаб}tj|tdГddНtj|dddН}tjа|dбtаб}tj|tdГddНtj|dddН}tjа|dбdS) Nrr rrzfloyd-warshall-numpyr5rr2) r$rerОrgrИrЮ┌testing┌assert_almost_equalrfrПrrr ┌test_specified_methods_numpyls z?TestAverageShortestPathLengthNumpy.test_specified_methods_numpyN)rАrБrВrГr/rбrrrr rЫds rЫ)r<r#r$rrrДrЫrrrr ┌s G