o  h$@s<ddlZddlZddlmZGdddZGdddZdS)N)convert_node_labels_to_integersc@sleZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZdS) TestCliquescCslgd}ttj|dd|_tt|j|_td}t |ddt dD}| gd||_ dS)N) rrrrrrrr) first_labelcSsi|]}||dqSr).0ir r Y/var/www/vscode/kcb/lib/python3.10/site-packages/networkx/algorithms/tests/test_clique.py sz,TestCliques.setup_method..))rr )r)rr)rr)rr) cnltinx generatorshavel_hakimi_graphGlist find_cliquesclcomplete_graph relabel_nodesrangeremove_edges_fromH)selfzrr r r setup_methods  zTestCliques.setup_methodcCstt|j}t|j}gdgdgdddgddgg}ttt|ttt|ks/Jttt|ttt|ks?JdS)Nrr rrrr r)rr )rrrrfind_cliques_recursivesortedmaprrrclexpectedr r rtest_find_cliques1s  " $zTestCliques.test_find_cliques1cs|jddtt|j}tt|j}ttt|ttt|ks'Jhdhdhdddhddhgt t |ksBJt fd d |DsOJdS) Nr>rrrr >rrr >rrr#r$r%r&r'c3s|] }t|vVqdSN)set)r canswerr r sz-TestCliques.test_selfloops..) radd_edgerrrr(r0r* frozensetlenall)rrr,r r2rtest_selfloopss "zTestCliques.test_selfloopscCsBtt|j}ttt|ddggdddggdgksJdS)Nrr)rrrr r)rrr )rrrrr)r*)rhclr r rtest_find_cliques2!s2zTestCliques.test_find_cliques2cCsTtt|jdg}t|jdg}gdgdg}ttt|ttt|ks*Jttt|ttt|ks:Jtt|jddg}t|jddg}gdg}ttt|ttt|kscJttt|ttt|kssJtt|jgd}t|jgd}gdg}ttt|ttt|ksJttt|ttt|ksJtt|jgd}t|jgd}gdg}ttt|ttt|ksJttt|ttt|ksJtt tt|jgdWdn 1swYtt tt|jgdWddS1s#wYdS)Nrr!r"r)rr rr) rrrrr(r)r*pytestraises ValueErrorr+r r rtest_find_cliques3%s4             $zTestCliques.test_find_cliques3c Csx|j}t|ddks Jtt|dgdgksJtt|ddgddgks/Jt|ddgdddks>Jt|ddksHJt|dddddddddddd ks]Jtj|t|ddddddddddddd ksvJtj|gddddddksJtj||jddddddddddddd ksJtj|t||jddddddddddddd ksJdS)Nrrrr rrrrrr r#r$r%r&r')nodes)rrrcliques)rrnumber_of_cliquesrvaluesrrrr r rtest_number_of_cliquesFsv $   "   z"TestCliques.test_number_of_cliquesc Cs,|j}t|ddks Jtt|dgdgksJtt|ddgddgks/Jt|ddgdddks>Jt|ddksHJt|dddddddddddd ks]Jtj||jddddddddddddd ksuJtj|ddg|jddddksJtj|d|jddksJdS)Nrrrr@rrArC)rrnode_clique_numberrrFrrGr r rtest_node_clique_numbersD $   $ z#TestCliques.test_node_clique_numbercCs|j}t|}t|gdksJt|tdd}|j|jks#Jt|tdd}t|ddtddD}t|gd ksDJdS) N)rrrrrr r#r$r%r&r'r rKrcSsi|]}| |qSr r r vr r rrsz:TestCliques.test_make_clique_bipartite..r )rrrrr)rrmake_clique_bipartiter)projected_graphradjr)rrBrH1r r rtest_make_clique_bipartites z&TestCliques.test_make_clique_bipartitecCs\|j}t|}t|tdd}t|ddtddD}t|}|j|jks,JdS)zTests that the maximal clique graph is the same as the bipartite clique graph after being projected onto the nodes representing the cliques. rKrcSsi|]}| |dqSr r rQr r rrsz:TestCliques.test_make_max_clique_graph..rr N)rrrSrTrrmake_max_clique_graphrU)rrrVrWH2r r rtest_make_max_clique_graphs   z&TestCliques.test_make_max_clique_graphcCsDttjtttWddS1swYdSr/)r<r=rNetworkXNotImplementednextrDiGraph)rr r r test_directeds"zTestCliques.test_directedcCs8t}tt|gksJtt|gksJdSr/)rGraphr)rr(rGr r rtest_find_cliques_trivialsz%TestCliques.test_find_cliques_trivialcCsFtgd}tgd}|dttj|tjd|s!JdS)N)r@)rr)rr)rr ))rr)rrr@r) create_using)rr`add_node is_isomorphicrY)rrEr r r'test_make_max_clique_graph_create_usings  z3TestCliques.test_make_max_clique_graph_create_usingN)__name__ __module__ __qualname__r r.r9r;r?rHrJrXr[r_rarfr r r rrs  !=$  rc@seZdZddZdS)TestEnumerateAllCliquescCst}gd}||tt|}ttt|}t||ks#Jgdgdgdgdgdgdgdgddggd gd gd ddggd gd gdddggdddgddggdgdgdgdddggdddgddgddggdgdgdddggdddggdddgddggdddgddgddggdgdgdgd}ttt|ttt|ksJdS)N))ab)rkr1)rkd)rke)rlr1)rlrm)rlrn)r1rm)r1rn)rmrn)frl)ror1)rog)rpro)rpr1)rprm)rprnrkrlr1rmrnrorp)rkrlrm)rkrlrmrn)rkrlrn)rkr1rm)rkr1rmrn)rkr1rn)rkrmrn)rlr1rm)rlr1rmrn)rlr1rn)rlr1ro)rlrmrn)r1rmrn)r1rmrnrp)r1rmrp)r1rnrp)r1rorp)rmrnrp)rkrlr1)rkrlr1rm)rkrlr1rmrn)rkrlr1rn)rr`add_edges_fromrenumerate_all_cliquesr*r7r))rr edges_fig_4rD clique_sizesexpected_cliquesr r rtest_paper_figure_4s       !"#$%&'()*+,-$0z+TestEnumerateAllCliques.test_paper_figure_4N)rgrhrirvr r r rrjs rj)r<networkxrrrrrjr r r rs  N