o &?eYI@sddlmZddlZddlZddlmZddlmZddlm Z GdddeZ Gd d d e e Z Gd d d Z Gd ddeZGdddejZGddde ZdS))UserDictN) edges_equal)BaseAttrGraphTester) TestGraphc@s|eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZdS)BaseMultiGraphTestercCsN|j}|dds J|ddrJ|dddsJ|dddr%JdS)Nrr)K3Zhas_edgeselfGr g/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/networkx/classes/tests/test_multigraph.py test_has_edge s z"BaseMultiGraphTester.test_has_edgecCs||j}|dddiiksJ|dddiiksJ|dddiks'J|dddus1J|dddiks|j}||t|}||||}|||dSN)r add_attributesnx MultiGraphis_shallow_copyZ to_undirected is_deepcopyr r r r r rtest_to_undirected^    z'BaseMultiGraphTester.test_to_undirectedcCr3r4)r r5r6Z MultiDiGraphr8Z to_directedr9r:r r rtest_to_directedfr<z%BaseMultiGraphTester.test_to_directedcCsv|j}|dd|dd|jdddd|jdddd|dddks)J|dd|dddks9JdS)Nr parallel edgekeyrr)r add_edge remove_edgenumber_of_edgesr r r rtest_number_of_edges_selfloopsns   z3BaseMultiGraphTester.test_number_of_edges_selfloopscCs^|}|jdddd|jdddddt|jdddis!Jt|jd ddis-JdS) Nrrbar)rr@bizrrrr)rrr@)GraphrAredgesr r r rtest_edge_lookupxs z%BaseMultiGraphTester.test_edge_lookupcs|jdddddjdddddtddjs"Jtfdd jd d Ds3Jtjd d d dddd difdddd difgsMJtjd d d ddgs[JdS)Nrrk1rE)r@rk2r+c3s"|] \}}}t|jVqdSr4) isinstanceedge_attr_dict_factory).0uvdr r r s z6BaseMultiGraphTester.test_edge_attr..TdatakeysrVr)rrrKrE)rrrLr+)rHrArMredge_key_dict_factoryallrIrr r rSrtest_edge_attrs     z#BaseMultiGraphTester.test_edge_attrc CsP|}|jdddddddt|jdd dddddd fgs"Jd |dddd <t|jdd ddd ddd fgs?Jd |jdddd <t|jdd ddd ddd fgs]Jd|jdd <t|jdd dddddd fgswJd dg|jdddd<d |jdddd<t|jdd dddddd dgd dfgsJdS)NrrrrEr)r@rVspamrETrU)rVr^rErrVrrGlistdataweight)rVr^rErarb)rHrArrIr-r r r rtest_edge_attr4s@  z$BaseMultiGraphTester.test_edge_attr4N)__name__ __module__ __qualname__rrrr!r#r)r0r2r;r=rDrJr\rcr r r rr s  rc@seZdZddZddZddZddd Zd d d Zeed ZddeiiZ ddeiiZ ddeddiiZ ddgiZ dddefdddefgZ dddifgZdddefgZdddefgZdddeddfgZe de fe defe defe defe defgZejdeddZe e fe efe efgZejdeddZe e e gZejdeddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Z d+d,Z!d-d.Z"d/d0Z#d1S)2TestMultiGraphcCstj|_diidiidii}}}||d||d||dd|_gd|_gd|_||_|j|j_i|j_i|jjd<i|jjd<i|jjd<dS) Nrrrrrrrrrr) r6r7rHZk3adjk3edgesk3nodesr r$r%)r Zed1Zed2Zed3r r r setup_methods       zTestMultiGraph.setup_methodcCs`|jdgdgddd}|jdksJdddiiifdddiiifg}t|j|ks.JdS)Nrrrtest)r'r)rHr'sortedr-items)r r expectedr r rtest_data_inputs zTestMultiGraph.test_data_inputc Cs0ddd}ddd}||d}dd|ii}ddd |fddd |fg}ddd |fg}|j|d d }t|jd d d |ksr?rrrrrr)r rArBr-rrr6rrr r rtest_remove_multiedges $"z$TestMultiGraph.test_remove_multiedgeN)$rdrerfrkrpr~ZetraitsZ egraphicsZedataZdodod1Zdodod2Zdodod3Zdolr{r|Z single_edge1Z single_edge2Z single_edge3casesrmarkZ parametrizerZmgi_none_casesrZ raise_casesrrrrrrrrrrr r r rrgsT                !  rgc@sPeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS)TestEdgeSubgraphz;Unit tests for the :meth:`MultiGraph.edge_subgraph` method.cCst}t|tdt|tdtdD] }d||j|d<qd|jdddd<d|jdddd<d|jd d dd<d |jd d dd<d |jd<||_|d dg|_ dS)Nrnoder'edge010rrZedge011Zedge340rredge341r&r)rrr) r6r7Zadd_pathranger.r-r&r Z edge_subgraphr )r r ir r rrks  zTestEdgeSubgraph.setup_methodcCsgdt|jks JdS)z.Tests that the subgraph has the correct nodes.rrrrN)rmr r.r[r r rtest_correct_nodessz#TestEdgeSubgraph.test_correct_nodescCs$ddgt|jjdddksJdS)z.Tests that the subgraph has the correct edges.)rrrr)rrrrTr'rWN)rmr rIr[r r rtest_correct_edgessz#TestEdgeSubgraph.test_correct_edgescC*|jdgdt|jksJdS)zkTests that adding a node to the original graph does not affect the nodes of the subgraph. rrN)r add_nodermr r.r[r r r test_add_node zTestEdgeSubgraph.test_add_nodecCr)ziTests that removing a node in the original graph does affect the nodes of the subgraph. r)rrrN)r rrmr r.r[r r rrrz!TestEdgeSubgraph.test_remove_nodecCs|jD]}|jj||jj|ksJqd|jjdd<|jjd|jjdks*Jd|jjdd<|jjd|jjdks@JdS)z`Tests that the node attribute dictionary of the two graphs is the same object. rrr'rErN)r r r.)r rQr r rtest_node_attr_dicts  z$TestEdgeSubgraph.test_node_attr_dictcCs|jjddD]\}}}|jj||||jj|||ks"Jqd|jjdddd<|jjdddd|jjddddksIJd|jjdd dd<|jjdd dd|jjdd ddksoJd S) z`Tests that the edge attribute dictionary of the two graphs is the same object. T)rXrrrr'rErrN)r rIr r$)r rPrQkr r rtest_edge_attr_dicts .48z$TestEdgeSubgraph.test_edge_attr_dictcCs|jj|jjus JdS)zaTests that the graph attribute dictionary of the two graphs is the same object. N)r r&r r[r r rtest_graph_attr_dictsz%TestEdgeSubgraph.test_graph_attr_dictN) rdrerf__doc__rkrrrrrrrr r r rrs rc@s eZdZdS)CustomDictClassN)rdrerfr r r rrsrc@s(eZdZeZeZeZeZeZeZ eZ dS)MultiGraphSubClassN) rdrerfrnode_dict_factorynode_attr_dict_factoryadjlist_outer_dict_factoryadjlist_inner_dict_factoryrYrNZgraph_attr_dict_factoryr r r rrsrc@seZdZddZdS)TestMultiGraphSubclasscCst|_gd|_gd|_||_|j|j|j|jd|j_iiid|j_|jD] }|jD]}||krQdii}||jj||<||jj||<q7q2|j |j_ |j |jj d<|j |jj d<|j |jj d<dS)Nrhrrrr) rrHrirjr rrr$r(rr%r)r rPrQrRr r rrks.     z#TestMultiGraphSubclass.setup_methodN)rdrerfrkr r r rrs r) collectionsrrZnetworkxr6Znetworkx.utilsrZ test_graphrrZ _TestGraphrrgrrr7rrr r r rs    +dQ