o ˷e-@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_viewrvselfr]/var/www/ideatree/venv/lib/python3.10/site-packages/networkx/classes/tests/test_graphviews.pysetup szTestReverseView.setupcCTddl}|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_itercCs$tj}t}ttj|j|dSr+)r graphviewsGraphpytestraisesNetworkXNotImplementedr)rnxgr 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__rDrIrrGrrrH'rHr$zRM classz RMC classrC) r r add_edgerprint __class__copyr'has_edgerD)rMRMRMCrrGr test_subclass&s     zTestReverseView.test_subclassN) rLrMrNrr!r(r7r?rYrrrrr s  rc@s4eZdZddZddZddZddZd d Zd S) TestMultiReverseViewcCs4tjdtd|_|jddt|j|_dS)Nrr)r r MultiDiGraphr rQrrrrrrr:szTestMultiReverseView.setupcCrrrrrrrr!?r"z TestMultiReverseView.test_picklecCsdd|jjvsJd|jjvsJd|jjvsJd|jjvs Jd|jjvs(Jd|jjvs0JdS)N)r$r%r)r%r$r)r\r[rP)r[r\rPr&rrrrr(Hs z"TestMultiReverseView.test_containscCr*)Ncss |] \}}}|||fVqdSr+r)r0uvkrrrr2Qsz1TestMultiReverseView.test_iter..r3r5rrrr7Pr8zTestMultiReverseView.test_itercCs(tj}t|j}ttj|j|dSr+)r r9 MultiGraphr r;r<r=r)rr>MGrrrr?Ts z$TestMultiReverseView.test_exceptionsN)rLrMrNrr!r(r7r?rrrrrZ9s   rZcCstj}tdg}ttj|j|tjdWdn1s"wYtdg}ttj|j|tjdWddS1sGwYdS)N)rPr$r) r r9r r;r< NetworkXErrorgeneric_graph_viewrar])r>r rrrtest_generic_multitypeZs  "rec@r)TestToDirectedcCs@td|_t|j|_tjdtd|_t|j|_dSr)r r r to_directeddvrarbMdvrrrrres zTestToDirected.setupcCs |jrJ|jsJdSr+)r is_directedrhrrrr test_directedkzTestToDirected.test_directedcCDt|j}t|j}t|j|jjsJt|j|jjs JdSr+)r rgrhrirr')rddMddrrrtest_already_directedo  z$TestToDirected.test_already_directedcCsdddl}|j}|||d}|j|jksJ|j|jks J|j|jks(J|j|jks0JdSr)rrhrrr_succ_predr)rrrhpdvrrrr!uszTestToDirected.test_picklecCsDd|jjvsJd|jjvsJd|jjvsJd|jjvs JdSr#)r r'rhrrrrr(r)zTestToDirected.test_containscCs>dd|jjD}tt|jj|}t|jj|ksJdS)NcSsg|]}tt|qSrr,r/rrr sz,TestToDirected.test_iter..)r r'r4listrh)rrevdr6rrrr7szTestToDirected.test_iterN rLrMrNrrkrpr!r(r7rrrrrfd rfc@r)TestToUndirectedcCsHtjdtd|_t|j|_tjdtd|_t|j|_dSr) r r r DG to_undirecteduvr]MDGMuvrrrrrszTestToUndirected.setupcCs |jsJ|jrJdSr+)r{rjr}rrrrrkrlzTestToUndirected.test_directedcCrmr+)r r|r}rrr')ruuMuurrrrprqz&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_iterNrxrrrrrzryrzc@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{rarbr]r~r|GvrgDGvMGvMDGvreverseRvMRvgraphsr'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%r[r\r) r subgraphr ranger{rbr~rinduced_subgraphrvr)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%r[r\)rr)r[r\rr)rrPr$r%N) r restricted_viewr rrrrr'rTremove_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 OrderedGraphrrTtyperorigGr rrrrrtest_subgraph_copys  z$TestChainsOfViews.test_subgraph_copycCsHt|jgd}|}t|gdksJt|jgdks"JdS)Nr)r[r\r\r[r\r)rr\)r rr rgr4r'rrrrrrtest_subgraph_todirectedz*TestChainsOfViews.test_subgraph_todirectedcCsHt|jgd}|}t|gdksJt|jddgks"JdS)Nrrr)r rr r|rvr4r'rrrrtest_subgraph_toundirected rz,TestChainsOfViews.test_subgraph_toundirectedcCsR|jjdd}|gd}|}t|gdksJt|jddgks'JdS)NFrTrrr)r{rrr|rvr4r')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)Nrrrr[r\)r rTr edge_subgraphr|rvr4r')rr rrUSSGrrr'test_subgraph_edgesubgraph_toundirected$s z9TestChainsOfViews.test_subgraph_edgesubgraph_toundirectedcCP|j}|gd}|jdd}|jdd}t|dsJt|dr&JdSNrTas_viewFr)r rTrrrr rCSGDCSGrrrtest_copy_subgraph,   z$TestChainsOfViews.test_copy_subgraphcCrr)r{rTrrrrrrtest_copy_disubgraph4rz&TestChainsOfViews.test_copy_disubgraphcCrr)r~rTrrrrrrtest_copy_multidisubgraph<rz+TestChainsOfViews.test_copy_multidisubgraphcCrr)rbrTrrrrrrtest_copy_multisubgraphDrz)TestChainsOfViews.test_copy_multisubgraphcCs<t|j}|jjdksJ|jdd}|jjdksJdS)NOrderedMultiGraphTr)r rrrSrLrT)rr rrrtest_copy_of_viewLs  z#TestChainsOfViews.test_copy_of_viewcsxGfdddtj|jD]+}|}|gd}|}|dks'J|dks/Jd|vs9d|vs9JqdS)Ncr@)z0TestChainsOfViews.test_subclass..MyGraphcSrArBrrrrrrDTrEz:TestChainsOfViews.test_subclass..MyGraph.my_methodcrFr+rrrGrrrIWrJzBTestChainsOfViews.test_subclass..MyGraph.to_directed_classNrKrrGrrrHSrOrHrrCr%)r r rrrTrDrrrGrrYRs zTestChainsOfViews.test_subclassN)rLrMrN classmethodrr!rrrrrrrrrrrrrrYrrrrrs$    r) r;networkxr networkx.utilsrrrrZrerfrzrrrrrs0! '&