o 3ήc.@sddlZddlZddlZejddZejddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3ZdS)4NcCst}|jddd|jddd|jddd|jdd d|jddd d d |jddd dd |jddddd |jddddd |S)Nardemandbcidiweightcapacity  nxDiGraphadd_nodeadd_edgeGrY/tmp/pip-target-vg8gfxp4/lib/python/networkx/algorithms/flow/tests/test_networksimplex.pysimple_flow_graphsrcCst}|jddd|jddd|jddddd |jdd dd |jdd d d |jd ddd |jd ddd |jddddd |SNsrrtrrrr r rr rrrrrrrsimple_no_flow_graphsr$cCsHd}|D]}||D]}|||||||d7}qq|S)z1Returns flow cost calculated from flow dictionaryrr )keys)rflowDictflowCostuvrrrget_flowcost_from_flowdict$s  "r*cCs6|}td}t|dd|iittjtj|dSNinfrrfloatrset_node_attributespytestraises NetworkXErrornetwork_simplexrrr,rrrtest_infinite_demand_raise-sr5cCs8|}td}t|dd| iittjtj|dSr+r-r4rrrtest_neg_infinite_demand_raise4sr6cCs<|}td}t|d|id|idttjtj|dS)Nr,r )rr)rr)r.rset_edge_attributesr0r1r2r3r4rrrtest_infinite_weight_raise;s r9cC.|}t|dddiittjtj|dS)Nrrrr/r0r1NetworkXUnfeasibler3rrrrrtest_nonzero_net_demand_raiseDr?cCs4|}t|ddiddidttjtj|dS)Nr rr ir7)rr8r0r1r=r3r>rrrtest_negative_capacity_raiseJsrAcCs|}ttjtj|dSN)r0r1rr=r3r$rrrrtest_no_flow_satisfying_demandsPsrDcCr:)Nr rr r<rCrrrtest_sum_demands_not_zeroUr@rEc Ct}gd}gd}gd}gd}gd}d}tt|D] }|j|d||dq tt|D]}|j||||||||d q4t|\}} ||ksUJ|t|| ks^Jd S) E https://developers.google.com/optimization/flow/mincostflow ) rrrrrrrr r ) rrrr r r r r r) r rrHr rJr) r r rrrrr rr )rJrrrrr Nrrrangelenrrr3r* r start_nodes end_nodes capacities unit_costssuppliesanswerir'r&rrrtest_google_or_tools_example[  rYc CrF) rG) rrrrrrrr r r ) rrrr r r r r rr) rHrIrJr rrHr rJrr) r r rrrrr rr r )rrrrKrMrr NrNrQrrrtest_google_or_tools_example2trZr^cCsRtjtjtd}t|}t|\}}d|ksJdt||ks'JdS)Nznetgen-2.gpickle.bz2l[$) ospathjoindirname__file__r read_gpickler3 cost_of_flow)fnamerr'r&rrr test_larges   rgcCst}|jddd|jddd|jddddd |jdd d d d |jddd dd |jd dddd t|\}}dd dddidd iid}|dksQJt|dksZJ||ks`JdS)Nrrrrrrr r r rrrrrr)rr)rrrr)rrrrr3min_cost_flow_costrr'Hsolnrrrtest_simple_digraphs rmcCst}|jddd|jddd|jddddd |jdd dd |jd dd d |jd ddd |jdd dd |jddddd ttjtj|dSr)rrrrr0r1r=r3rrrrtest_negcycle_infcapsrnc Cst}|jddd|jddd|jddd|jddd|jd d d|jd d d|jd dd|jd dd|jddd|jdddd|jdddd|jdddd|jdddd|jdddd|jdddd|jd ddd|jd d dd|jd ddd|jd ddd|jd d dd|jd ddd|jd d dd|jd d ddt|\}}ddidddddiidddddddddiddddddd }|d ksJ||ksJdS)!Nrrrrr#rrr er;fghrrr!rrr"rr  rMi)rrs)rsrp)rrqrr)rrq)rr) rrrrrorprqrrrs)rrrrr3rjrrrtest_transshipmentsJ  rxcCst}|jddd|jddd|jddd|dddddfdd d ddfdd d difdddddfddd d dfd ddddfd ddd dfddd difdd dddfg t|\}}dd dd dddddddd id did}|dks~Jt|dksJ||ksJdS)Nrirr rrHrr r r rIr rrru)rr r)r r r )r rrrr r rrL)rrradd_edges_fromr3rirjrrr test_digraph1s4     r}c Cst}|dddddfdddddfdddddfdddddfdddddfdddddfdddddfgd|jdd <d|jdd <d|jdd <t|\}}ddd ddd ddiiddd d }|dksoJt|dksxJ||ks~JdS)z+Address issue raised in ticket #617 by arv.rrryrr rr rMr)rr)r r)r r r{rN)rrr|nodesr3rirjrrrtest_zero_capacity_edgess& & rcCsdifdddifdddifg}dddddfdddd dfddd d dfdddd dfg}t|}||t|\}}dd id dd dd id }|dksQJdS)zHCheck if digons are handled properly. Taken from ticket #618 by arv.rrrr;r r i' ryrr- )rr rrr i+N)rradd_nodes_fromr3)r~edgesrr'rkrlrrr test_digons  rcCs`t}|jtdddd|jdd<d|jdd<|jgd d d d ttjtj |d S)zZCheck if one-node cycles are handled properly. Taken from ticket #2906 from @sshraven.rrrir r r ))rr)rr )rrrJg?ryN) rrrrOr~r|r0r1r=r3rrrr test_deadend%s rcCsdifdddifdddifg}ddddifdddd ifddd d d fdddd d fg}t|}||ttjtj|d S)zNAn infinite capacity negative cost digon results in an unbounded instance.rrrr;r r r irrrryN)rrrr0r1NetworkXUnboundedr3)r~rrrrr test_infinite_capacity_neg_digon2s    rcCs\t}|jddgddt|\}}|dksJ|dddiidddiiidks,Jd S) zMultidigraphs are acceptable.)rrr)rr rr r!rrr rN)r MultiDiGraphadd_weighted_edges_fromr3rr'rkrrrtest_multidigraphAs  (rcCst}|jddddttjtj|d|ddd<t|\}}|dks*J|dddiiks4Jt}|jddddd|jdddddttjtj|d|dddd<t|\}}|dksjJ|dddd d iikswJd S) ziNegative selfloops should cause an exception if uncapacitated and always be saturated otherwise. rrMr!rr r#xyr)rrN)rrrr0r1rr3rrrrrtest_negative_selfloopsJs  rcCst}|jddd|jddd|jddd|jddd|jddd|jd dd|jdddd |jdddd |jdddd |jd ddd |jdddd t|\}}|dkscJ|dddd iiiddiddid ksxJdS) Nrr;rrrr r r#r)r r)rrrr r rrwrrrrtest_bone_shapedbs .rcCsRt}ttjtj|t}ttjtj|t}ttjtj|dSrB) rGraphr0r1NetworkXNotImplementedr3 MultiGraphrr2rrrrtest_graphs_type_exceptionsus r)r_r0networkxrfixturerr$r*r5r6r9r?rArDrErYr^rgrmrnrxr}rrrrrrrrrrrrs<      )!