U ╣mЬd-у@srddlZddlZddlmZmZGddДdГZGddДdГZddДZGd d Дd ГZ GddДdГZ Gd dДdГZdS)щN)┌edges_equal┌nodes_equalc@s<eZdZddДZddДZddДZddДZd d ДZddДZd S)┌TestReverseViewcCs&tjdtабdН|_tа|jб|_dSйNщ йZcreate_using)┌nx┌ path_graph┌DiGraph┌G┌reverse_view┌rvй┌selfйr·_/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/networkx/classes/tests/test_graphviews.py┌setup_method szTestReverseView.setup_methodcCsTddl}|j}|а|а|dбб}|j|jks0tВ|j|jks@tВ|j|jksPtВdSйNrщ й┌pickler ┌loads┌dumps┌_node┌AssertionError┌_adj┌graphйrrr Zprvrrr┌test_pickleszTestReverseView.test_picklecCsDd|jjkstВd|jjks tВd|jjks0tВd|jjks@tВdSйN)щщ)r!r йr┌edgesrr rrrr┌ test_containsszTestReverseView.test_containscCs.tddД|jjDГГ}t|jjГ|ks*tВdS)Ncss|]}tt|ГГVqdSйNй┌tuple┌reversedй┌.0┌errr┌ sz,TestReverseView.test_iter..й┌sortedrr#r rйr┌expectedrrr┌ test_iterszTestReverseView.test_itercCs$tj}tаб}tаtj|j|бdSr%)r┌ graphviews┌Graph┌pytest┌raises┌NetworkXNotImplementedr)r┌nxgrrrr┌test_exceptions!szTestReverseView.test_exceptionscsАGЗfddДdtjГЙИГ}|аddбtа|б}td|jГ|аб}td|jГt|jГ|аddбslt В|а бdks|t ВdS)Ncs eZdZddДZЗfddДZdS)z.TestReverseView.test_subclass..MyGraphcSsdSйN┌merrrrr┌ my_method(sz8TestReverseView.test_subclass..MyGraph.my_methodcsИГSr%rrй┌MyGraphrr┌to_directed_class+sz@TestReverseView.test_subclass..MyGraph.to_directed_classNй┌__name__┌ __module__┌__qualname__r;r>rr<rrr='sr=щr zRM classz RMC classr:)rr ┌add_edger┌print┌ __class__┌copyr#Zhas_edgerr;)r┌MZRMZRMCrr<r┌ test_subclass&s zTestReverseView.test_subclassN) r@rArBrrr$r1r8rIrrrrr s rc@s4eZdZddДZddДZddДZddДZd d ДZdS)┌TestMultiReverseViewcCs4tjdtабdН|_|jаddбtа|jб|_dS)Nrrщщ)rr ┌MultiDiGraphrrDrr rrrrr:sz!TestMultiReverseView.setup_methodcCsTddl}|j}|а|а|dбб}|j|jks0tВ|j|jks@tВ|j|jksPtВdSrrrrrrr?sz TestMultiReverseView.test_picklecCsdd|jjkstВd|jjks tВd|jjks0tВd|jjks@tВd|jjksPtВd|jjks`tВdS)N)r r!r)r!r r)rLrKrC)rKrLrCr"rrrrr$Hsz"TestMultiReverseView.test_containscCs.tddД|jjDГГ}t|jjГ|ks*tВdS)Ncss|]\}}}|||fVqdSr%r)r*┌u┌v┌krrrr,Qsz1TestMultiReverseView.test_iter..r-r/rrrr1PszTestMultiReverseView.test_itercCs(tj}tа|jб}tаtj|j|бdSr%)rr2┌ MultiGraphrr4r5r6r)rr7┌MGrrrr8Tsz$TestMultiReverseView.test_exceptionsN)r@rArBrrr$r1r8rrrrrJ9s rJc Csrtj}tаdgб}tаtjбП|j|tjdНW5QRXtаdgб}tаtjбП|j|tjdНW5QRXdS)N)rCr r) rr2r r4r5Z NetworkXErrorZgeneric_graph_viewrQrM)r7rrrr┌test_generic_multitypeZsrSc@s<eZdZddДZddДZddДZddДZd d ДZddДZd S)┌TestToDirectedcCs@tаdб|_tа|jб|_tjdtабdН|_tа|jб|_dSr)rr r┌to_directed┌dvrQrR┌MdvrrrrreszTestToDirected.setup_methodcCs |jабrtВ|jабstВdSr%)r┌is_directedrrVrrrr┌ test_directedkszTestToDirected.test_directedcCsDtа|jб}tа|jб}t|j|jjГs,tВt|j|jjГs@tВdSr%)rrUrVrWrr#r)r┌ddZMddrrr┌test_already_directedosz$TestToDirected.test_already_directedcCsdddl}|j}|а|а|dбб}|j|jks0tВ|j|jks@tВ|j|jksPtВ|j|jks`tВdSr) rrVrrrrZ_succZ_predr)rrrVZpdvrrrruszTestToDirected.test_picklecCsDd|jjkstВd|jjks tВd|jjks0tВd|jjks@tВdSr)rr#rrVrrrrr$szTestToDirected.test_containscCs>ddД|jjDГ}tt|jjГ|Г}t|jjГ|ks:tВdS)NcSsg|]}tt|ГГСqSrr&r)rrr┌ Жsz,TestToDirected.test_iter..)rr#r.┌listrVr)rZrevdr0rrrr1ЕszTestToDirected.test_iterNй r@rArBrrYr[rr$r1rrrrrTds rTc@s<eZdZddДZddДZddДZddДZd d ДZddДZd S)┌TestToUndirectedcCsHtjdtабdН|_tа|jб|_tjdtабdН|_tа|jб|_dSr) rr r ┌DG┌ to_undirected┌uvrM┌MDG┌MuvrrrrrМszTestToUndirected.setup_methodcCs |jабstВ|jабrtВdSr%)r`rXrrbrrrrrYТszTestToUndirected.test_directedcCsDtа|jб}tа|jб}t|j|jjГs,tВt|j|jjГs@tВdSr%)rrarbrdrr#r)r┌uuZMuurrrr[Цsz&TestToUndirected.test_already_directedcCsbddl}|j}|а|а|dбб}|j|jks0tВ|j|jks@tВ|j|jksPtВt|dГs^tВdS)Nrr┌_graph) rrbrrrrrr┌hasattr)rrrbZpuvrrrrЬszTestToUndirected.test_picklecCsDd|jjkstВd|jjks tВd|jjks0tВd|jjks@tВdSr)r`r#rrbrrrrr$жszTestToUndirected.test_containscCs$t|jjГ}t|jjГ|ks tВdSr%)r.r`r#rbrr/rrrr1мszTestToUndirected.test_iterNr^rrrrr_Лs r_c@sРeZdZeddДГZddДZddДZddДZd d ДZddДZ d dДZ ddДZddДZddДZ ddДZddДZddДZddДZddДZdd ДZd!S)"┌TestChainsOfViewsc Csцtаdб|_tjdtабdН|_tjdtабdН|_tjdtабdН|_tа |jб|_ tа|jб|_tа |jб|_ tа|jб|_|jаб|_|jаб|_|j|j|j|j|j |j|j |j|j|jg |_|jD]}|j|j|jfq╠dSr)rr rr r`rQrRrMrcraZGvrUZDGv┌MGvZMDGv┌reverseZRvZMRv┌graphsr#┌nodesZdegree)┌clsrrrr┌setup_class▓s.Ў zTestChainsOfViews.setup_classcCsNddl}|jD]:}|а|а|dбб}t|j|jГs6tВt|j|jГstВqdSr) rrkrrrr#rrrl)rrr┌Hrrrr═s zTestChainsOfViews.test_picklecCs┬tа|jtddГб}tа|jtddГб}tа|jtddГб}tа|jtddГб}|j||||gD]Z}tа|dddgб}t |ГdddgksОt В|аddgб}t |Гdgksоt В|j|ksbt ВqbdS)Nr!щrKrLщ)r┌subgraphr┌ranger`rRrcrk┌induced_subgraphr]rrf)rZSGvZSDGvZSMGvZSMDGvr┌SG┌SSGrrr┌test_subgraph_of_subgraph╒sz+TestChainsOfViews.test_subgraph_of_subgraphc Cs dddg}dg}tа|j||б}dddddg}tа||б}|а|б}|j|jksTtВ|j|jksdtВ|j|ksrtВt|j|jГsДtВ|jа б}|а |б|а|бt|а|б|jГs╕tВ|а dd d dgбt|j|jГs▄tВ|jа|б}tа|||б} | j|jk РstВt| j|jГРstВdS)a&Test subgraph chains that both restrict and show nodes/edges. A restricted_view subgraph should allow induced subgraphs using G.subgraph that automagically without a chain (meaning the result is a subgraph view of the original graph not a subgraph-of-subgraph. r!rKrL)rqrprqrpщrrCr N)rZrestricted_viewrrtrrrfrrr#rGZremove_nodes_fromZremove_edges_from) rZ hide_nodesZ hide_edgesZRGrlrurvZCGZSSSGZRSGrrr┌'test_restricted_induced_subgraph_chainsтs( z9TestChainsOfViews.test_restricted_induced_subgraph_chainscCsF|jD]:}tа|б}|аdddgб}|аб}t|Гt|ГkstВqdS)NrKrLrq)rkrr3rrrG┌typerйrZorigGrrurorrr┌test_subgraph_copys z$TestChainsOfViews.test_subgraph_copycCsPtа|jdddgб}|аб}t|Гdddgks2tВt|jГddddgksLtВdS)NrKrLrqйrKrLйrLrKйrLrq)rqrL)rrtrrUr.rr#йrrurvrrr┌test_subgraph_todirectedsz*TestChainsOfViews.test_subgraph_todirectedcCsLtа|jdddgб}|аб}t|Гdddgks2tВt|jГddgksHtВdS)NrKrLrqr}r)rrtrrar]rr.r#rАrrr┌test_subgraph_toundirected sz,TestChainsOfViews.test_subgraph_toundirectedcCsV|jjddН}|аdddgб}|аб}t|ГdddgkstВ|jjddН}|jddН}|j|jjksjtВ|j|jjks|tВdS)NFrГT)r`rjrlrr#rc)rrrorrr┌test_reverse_reverse_copysz+TestChainsOfViews.test_reverse_reverse_copycCs\|jаб}|аdddgб}|аddgб}|аб}t|ГddgksDtВt|jГdgksXtВdS)NrKrLrqr}r~) rrGrrZ edge_subgraphrar]rr.r#)rrrurvZUSSGrrr┌'test_subgraph_edgesubgraph_toundirected$s z9TestChainsOfViews.test_subgraph_edgesubgraph_toundirectedcCsR|jаб}|аdddgб}|jddН}|jddН}t|dГs@tВt|dГrNtВdSйNrKrLrqTйZas_viewFrf)rrGrrrgrйrrruZCSGZDCSGrrr┌test_copy_subgraph,s z$TestChainsOfViews.test_copy_subgraphcCsR|jаб}|аdddgб}|jddН}|jddН}t|dГs@tВt|dГrNtВdSrЗ)r`rGrrrgrrЙrrr┌test_copy_disubgraph4s z&TestChainsOfViews.test_copy_disubgraphcCsR|jаб}|аdddgб}|jddН}|jddН}t|dГs@tВt|dГrNtВdSrЗ)rcrGrrrgrrЙrrr┌test_copy_multidisubgraph<s z+TestChainsOfViews.test_copy_multidisubgraphcCsR|jаб}|аdddgб}|jddН}|jddН}t|dГs@tВt|dГrNtВdSrЗ)rRrGrrrgrrЙrrr┌test_copy_multisubgraphDs z)TestChainsOfViews.test_copy_multisubgraphcCs<tа|jб}|jjdkstВ|jddН}|jjdks8tВdS)NrQTrИ)rrQrirFr@rrG)rrrrr┌test_copy_of_viewLsz#TestChainsOfViews.test_copy_of_viewcszGЗfddДdtjГЙ|jD]X}И|Г}|аdddgб}|аб}|абdksPtВ|абdks`tВd|ksd|kstВqdS)Ncs eZdZddДZЗfddДZdS)z0TestChainsOfViews.test_subclass..MyGraphcSsdSr9rrrrrr;Tsz:TestChainsOfViews.test_subclass..MyGraph.my_methodcsИГSr%rrr<rrr>WszBTestChainsOfViews.test_subclass..MyGraph.to_directed_classNr?rr<rrr=Ssr=rKrLrqr:r!)rr rkrrrGr;rr{rr<rrIRs zTestChainsOfViews.test_subclassN)r@rArB┌classmethodrnrrwryr|rБrВrДrЕrЖrКrЛrМrНrОrIrrrrrh▒s" rh)r4ZnetworkxrZnetworkx.utilsrrrrJrSrTr_rhrrrr┌s0! '&