o  h,@srddlZddlZddlmZmZGdddZGdddZddZGd d d Z Gd d d Z Gd ddZ dS)N) edges_equal nodes_equalc@<eZdZddZddZddZddZd d Zd d Zd S)TestReverseViewcCs&tjdtd|_t|j|_dSN  create_using)nx path_graphDiGraphG reverse_viewrvselfrZ/var/www/vscode/kcb/lib/python3.10/site-packages/networkx/classes/tests/test_graphviews.py setup_method szTestReverseView.setup_methodcCTddl}|j}|||d}|j|jksJ|j|jks J|j|jks(JdSNrpicklerloadsdumps_node_adjgraphrrrprvrrr test_pickle zTestReverseView.test_picklecCsDd|jjvsJd|jjvsJd|jjvsJd|jjvs JdSN))r%r$r edgesrrrrr test_containszTestReverseView.test_containscC.tdd|jjD}t|jj|ksJdS)Ncss|] }tt|VqdSNtuplereversed.0errr sz,TestReverseView.test_iter..sortedr r'rrexpectedrrr test_iterzTestReverseView.test_itercCst}ttjtj|dSr+)r GraphpytestraisesNetworkXNotImplementedrrr rrrtest_exceptions!szTestReverseView.test_exceptionscsGfdddtj}|ddt|}td|j|}td|jt|j|dds6J| dks>JdS)Nc eZdZddZfddZdS)z.TestReverseView.test_subclass..MyGraphcSdSNmerrrrr my_method'z8TestReverseView.test_subclass..MyGraph.my_methodcSr+rrMyGraphrrto_directed_class*z@TestReverseView.test_subclass..MyGraph.to_directed_classN__name__ __module__ __qualname__rCrHrrFrrrG&rGr$zRM classz RMC classrB) r r add_edgerprint __class__copyr'has_edgerC)rMRMRMCrrFr test_subclass%s     zTestReverseView.test_subclassN) rKrLrMrr!r(r7r>rXrrrrr s  rc@s4eZdZddZddZddZddZd d Zd S) TestMultiReverseViewcCs4tjdtd|_|jddt|j|_dS)Nrr)r r MultiDiGraphr rPrrrrrrr9sz!TestMultiReverseView.setup_methodcCrrrrrrrr!>r"z TestMultiReverseView.test_picklecCsdd|jjvsJd|jjvsJd|jjvsJd|jjvs Jd|jjvs(Jd|jjvs0JdS)N)r$r%r)r%r$r)r[rZrO)rZr[rOr&rrrrr(Gs z"TestMultiReverseView.test_containscCr*)Ncss |] \}}}|||fVqdSr+r)r0uvkrrrr2Psz1TestMultiReverseView.test_iter..r3r5rrrr7Or8zTestMultiReverseView.test_itercCs"t|j}ttjtj|dSr+)r MultiGraphr r:r;r<r)rMGrrrr>Ss z$TestMultiReverseView.test_exceptionsN)rKrLrMrr!r(r7r>rrrrrY8s   rYcCstj}tdg}ttj|j|tjdWdn1s"wYtdg}ttj|j|tjdWddS1sGwYdS)N)rOr$r) r graphviewsr r:r; NetworkXErrorgeneric_graph_viewr`r\)nxgr rrrtest_generic_multitypeXs  "rfc@r)TestToDirectedcCs@td|_t|j|_tjdtd|_t|j|_dSr)r r r to_directeddvr`raMdvrrrrrcs zTestToDirected.setup_methodcCs |jrJ|jsJdSr+)r is_directedrirrrr test_directedizTestToDirected.test_directedcCDt|j}t|j}t|j|jjsJt|j|jjs JdSr+)r rhrirjrr')rddMddrrrtest_already_directedm  z$TestToDirected.test_already_directedcCsdddl}|j}|||d}|j|jksJ|j|jks J|j|jks(J|j|jks0JdSr)rrirrr_succ_predr)rrripdvrrrr!sszTestToDirected.test_picklecCsDd|jjvsJd|jjvsJd|jjvsJd|jjvs JdSr#)r r'rirrrrr(}r)zTestToDirected.test_containscCs>dd|jjD}tt|jj|}t|jj|ksJdS)NcSsg|]}tt|qSrr,r/rrr sz,TestToDirected.test_iter..)r r'r4listri)rrevdr6rrrr7szTestToDirected.test_iterN rKrLrMrrlrqr!r(r7rrrrrgb rgc@r)TestToUndirectedcCsHtjdtd|_t|j|_tjdtd|_t|j|_dSr) r r r DG to_undirecteduvr\MDGMuvrrrrrszTestToUndirected.setup_methodcCs |jsJ|jrJdSr+)r|rkr~rrrrrlrmzTestToUndirected.test_directedcCrnr+)r r}r~rrr')ruuMuurrrrqrrz&TestToUndirected.test_already_directedcCsbddl}|j}|||d}|j|jksJ|j|jks J|j|jks(Jt|ds/JdS)Nrr_graph)rr~rrrrrhasattr)rrr~puvrrrr!szTestToUndirected.test_picklecCsDd|jjvsJd|jjvsJd|jjvsJd|jjvs JdSr#)r|r'r~rrrrr(r)zTestToUndirected.test_containscCs$t|jj}t|jj|ksJdSr+)r4r|r'r~r5rrrr7s zTestToUndirected.test_iterNryrrrrr{rzr{c@seZdZeddZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZddZddZdd Zd!S)"TestChainsOfViewsc Cstd|_tjdtd|_tjdtd|_tjdtd|_t |j|_ t |j|_ t |j|_ t |j|_|j|_|j|_|j|j|j|j|j |j |j |j|j|jg |_|jD] }|j|j|jfqfdSr)r r r r r|r`rar\rr}GvrhDGvMGvMDGvreverseRvMRvgraphsr'nodesdegree)clsr rrr setup_classs0    zTestChainsOfViews.setup_classcCsNddl}|jD]}|||d}t|j|jsJt|j|js$JqdSr)rrrrrr'rr)rrr Hrrrr!s  zTestChainsOfViews.test_picklecCst|jtdd}t|jtdd}t|jtdd}t|jtdd}|j||||gD]+}t|gd}t |gdksEJ|ddg}t |dgksUJ|j |us\Jq1dS)Nr%rZr[r) r subgraphr ranger|rarrinduced_subgraphrwr)rSGvSDGvSMGvSMDGvr SGSSGrrrtest_subgraph_of_subgraphsz+TestChainsOfViews.test_subgraph_of_subgraphc Csgd}dg}t|j||}gd}t||}||}|j|jus&J|j|jus.J|j|us5Jt|j|js>J|j}| || |t|||jsXJ| gdt|j|jshJ|j|}t|||} | j|jus}Jt| j|jsJdS)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%rZr[)rr)rZr[rr)rrOr$r%N) r restricted_viewr rrrrr'rSremove_nodes_fromremove_edges_from) r hide_nodes hide_edgesRGrrrCGSSSGRSGrrr'test_restricted_induced_subgraph_chainss(      z9TestChainsOfViews.test_restricted_induced_subgraph_chainscCsD|jD]}t|}|gd}|}t|t|ksJqdS)Nr)rr r9rrStyperorigGr rrrrrtest_subgraph_copys  z$TestChainsOfViews.test_subgraph_copycCsHt|jgd}|}t|gdksJt|jgdks"JdS)Nr)rZr[r[rZr[r)rr[)r rr rhr4r'rrrrrrtest_subgraph_todirectedz*TestChainsOfViews.test_subgraph_todirectedcCsHt|jgd}|}t|gdksJt|jddgks"JdS)Nrrr)r rr r}rwr4r'rrrrtest_subgraph_toundirected rz,TestChainsOfViews.test_subgraph_toundirectedcCsR|jjdd}|gd}|}t|gdksJt|jddgks'JdS)NFrSrrr)r|rrr}rwr4r')rr rrrrr"test_reverse_subgraph_toundirecteds z4TestChainsOfViews.test_reverse_subgraph_toundirectedcCs|jjdd}|jdd}|j|jjksJ|j|jjksJ|jjdd}|jdd}|j|jjks5J|j|jjks>JdS)NFrT)r|rrr'r)rr rrrrtest_reverse_reverse_copys  z+TestChainsOfViews.test_reverse_reverse_copycCsZ|j}|gd}|ddg}|}t|ddgks!Jt|jdgks+JdS)NrrrrZr[)r rSr edge_subgraphr}rwr4r')rr rrUSSGrrr'test_subgraph_edgesubgraph_toundirected"s z9TestChainsOfViews.test_subgraph_edgesubgraph_toundirectedcCP|j}|gd}|jdd}|jdd}t|dsJt|dr&JdSNrTas_viewFr)r rSrrrr rCSGDCSGrrrtest_copy_subgraph*   z$TestChainsOfViews.test_copy_subgraphcCrr)r|rSrrrrrrtest_copy_disubgraph2rz&TestChainsOfViews.test_copy_disubgraphcCrr)rrSrrrrrrtest_copy_multidisubgraph:rz+TestChainsOfViews.test_copy_multidisubgraphcCrr)rarSrrrrrrtest_copy_multisubgraphBrz)TestChainsOfViews.test_copy_multisubgraphcCs<t|j}|jjdksJ|jdd}|jjdksJdS)Nr`Tr)r r`rrRrKrSr=rrrtest_copy_of_viewJs  z#TestChainsOfViews.test_copy_of_viewcsxGfdddtj|jD]+}|}|gd}|}|dks'J|dks/Jd|vs9d|vs9JqdS)Ncr?)z0TestChainsOfViews.test_subclass..MyGraphcSr@rArrrrrrCRrDz:TestChainsOfViews.test_subclass..MyGraph.my_methodcrEr+rrrFrrrHUrIzBTestChainsOfViews.test_subclass..MyGraph.to_directed_classNrJrrFrrrGQrNrGrrBr%)r r rrrSrCrrrFrrXPs zTestChainsOfViews.test_subclassN)rKrLrM classmethodrr!rrrrrrrrrrrrrrXrrrrrs$    r) r:networkxr networkx.utilsrrrrYrfrgr{rrrrrs/  '&