o 3ήc~ @sXddlZddlZddlmZddZddZddZd d Zd d Z d dZ ddZ dS)N)maxcutcCs<||}|t|jksJt|t||ksJdS)N)unionsetnodeslennumber_of_nodes)Gset1set2rr Z/tmp/pip-target-vg8gfxp4/lib/python/networkx/algorithms/approximation/tests/test_maxcut.py _is_valid_cuts  r cCs:t|D]\}}tjj|||hdd}||ksJqdS)Nweightr) enumeratenx algorithmscut_size)rrr inodecut_size_without_noder r r _cut_is_locally_optimal s  rcCs0td}tj|dd\}\}}t|||dS)N)seed)rcomplete_graphrrandomized_partitioningr r_r r r r r test_random_partitionings rcCsTtd}tj|dd\}\}}t|||t||ks Jt|dks(JdS)Nr)pr)rrrrr rrrr r r #test_random_partitioning_all_to_ones  r!cCstd}td|jddD]\}}}tdddd|d<qttjt| dd }t j ||ddd \}\}}t |||t |||dS) NrTdataidr rkrr)rrrandomredges randrangersamplesortedrr one_exchanger rruvw initial_cutrr r r r r test_one_exchange_basic$s   r4cCst}|jdddd|jdddd|jdddd|jdddd|jddddtj|ddd\}\}}t|||t||||d ksJJdS) Nrrrrr()rGraphadd_edgerr.r r)rrr r r r r test_one_exchange_optimal3s  r;cCstd}td|jddD] \}}}dt|d<qttjt|dd}t j ||dd\}\}}t |||t |||t |t |jksWt |t |jksYJdSdS)NrTr"rr&r)rrr)rr*rr,r-rrr.r rrr/r r r test_negative_weightsDs    0r=) r)networkxr!networkx.algorithms.approximationrr rrr!r4r;r=r r r r s