fiCjddlZddlZddlmZmZmZGddZdZdZ GddZ dS) N)check_planarity_recursiveget_counterexampleget_counterexample_recursivecZeZdZdZeddZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZejdej ej!fdZ"ejdgddZ#dS) TestLRPlanaritya`Nose Unit tests for the :mod:`networkx.algorithms.planarity` module. Tests three things: 1. Check that the result is correct (returns planar if and only if the graph is actually planar) 2. In case a counter example is returned: Check if it is correct 3. In case an embedding is returned: Check if its actually an embedding Nc6tj|d\}}t|d\}}|'|rd}nd}||ks J|||ks J||r"t||t||dSt ||t ||dS)a,Raises an exception if the lr_planarity check returns a wrong result Parameters ---------- G : NetworkX graph is_planar : bool The expected result of the planarity check. If set to None only counter example or embedding are verified. TNz/Wrong planarity check result. Should be planar.z3Wrong planarity check result. Should be non-planar.)nxcheck_planarityrcheck_embeddingcheck_counterexample)G is_planar is_planar_lrresultis_planar_lr_rec result_recmsgs /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/networkx/algorithms/tests/test_planarity.py check_graphzTestLRPlanarity.check_graphs "1!T:: f'@D'I'I$*   LGK ,,,c,,, 0000#000  0 Av & & & Az * * * * * !F + + + J / / / / /c`gd}|tj|ddS)N) rrrr )r"rr)r$rrrrr$r$r"Trrr Graphselfes rtest_simple_planar_graphz(TestLRPlanarity.test_simple_planar_graph:s<  !55555rc`gd}|tj|ddS)N) )rrrr)rr)rr)r$r$rrrr#)rr$r%rrr$r&Tr(r)r+s rtest_planar_with_selfloopz)TestLRPlanarity.test_planar_with_selfloopJs<    !55555rcZ|tjddddS)NrFr()rr complete_bipartite_graphr,s r test_k3_3zTestLRPlanarity.test_k3_3\s. 4Q::eLLLLLrcX|tjdddSNr$Fr()rr complete_graphr6s rtest_k5zTestLRPlanarity.test_k5_s, *1--?????rc`gd}|tj|ddS)N)rrrrr&r$r )r rTr(r)r+s rtest_multiple_components_planarz/TestLRPlanarity.test_multiple_components_planarbs4 < < < !55555rctjd}|gd||ddS)Nr$)r!r")rBr Fr()r r:add_edges_fromrr,r s r#test_multiple_components_non_planarz3TestLRPlanarity.test_multiple_components_non_planarfsN  a  111222 e,,,,,rctjd}tdD]}|||||ddSr9)r r:rangeadd_edger)r,r is rtest_non_planar_with_selfloopz-TestLRPlanarity.test_non_planar_with_selfloopms]  a q  A JJq!     e,,,,,rc`gd}|tj|ddS)N) r#)rr rr"rr rr2)rr"r&rrr"Fr(r)r+s rtest_non_planar1z TestLRPlanarity.test_non_planar1ts<  !66666rcdddg}tj|}||ddS)Nrr0Tr(r r*rr,r-r s r test_loopzTestLRPlanarity.test_loops8 V  HQKK d+++++rcddg}tj|}|dd||ddS)NrrrrTr()r r* remove_edgerrRs r test_compzTestLRPlanarity.test_compsL V  HQKK a d+++++rcdgd}tj|}||ddS)N)rr1rrr#rL)rrB)r r rr%rM)rr")r )rrY)rr[rr&rrNr'r!rA)r"r\r"rY)rBrY)r\rY)rYr[Tr(rQrRs rtest_goldner_hararyz#TestLRPlanarity.test_goldner_hararysA   : HQKK d+++++rc`tjgd}||ddS)N)rrrrrr=Tr()r MultiGraphrrDs rtest_planar_multigraphz&TestLRPlanarity.test_planar_multigraphs7 MJJJ K K d+++++rctjtjd}|dgdz||ddS)Nr$rFr()r r`r:rCrrDs rtest_non_planar_multigraphz*TestLRPlanarity.test_non_planar_multigraphsU M"+A.. / / &A&&& e,,,,,rc`tjgd}||ddS)N)rrr%)rr)rrrX)rrTr()r DiGraphrrDs rtest_planar_digraphz#TestLRPlanarity.test_planar_digraphs7 JOOO P P d+++++rctjtjd}|dd|dd||ddS)Nr$rrrFr()r rer:rUrrDs rtest_non_planar_digraphz'TestLRPlanarity.test_non_planar_digraphsc Jr(++ , , a a e,,,,,rctj}|d||ddS)NrTr()r r*add_noderrDs rtest_single_componentz%TestLRPlanarity.test_single_components; HJJ 1  d+++++rc`tjgd}||ddS)N) )rrYr rrn)r"r[rrB)rBrnrrrr")rrYrLTr(rQrDs r test_graph1zTestLRPlanarity.test_graph1sC H     d+++++rc`tjgd}||ddS)N)r)rrn)rrnr&r]rLrrrMr>)r"rn)rrBrp)rr\rmr!)rr )rrBFr(rQrDs r test_graph2zTestLRPlanarity.test_graph2sC H     * e,,,,,rc`tjgd}||ddS)N)rr)rr[r)rBr\)rr[rLrorZr2r'r1)rr)r$r[)r$rnFr(rQrDs r test_graph3zTestLRPlanarity.test_graph3sC H     $ e,,,,,rctjtj5tj}|dt |ddddS#1swxYwYdSNr)pytestraisesr NetworkXExceptionr*rjrrDs rtest_counterexample_planarz*TestLRPlanarity.test_counterexample_planar s ]2/ 0 0 " " A JJqMMM q ! ! !  " " " " " " " " " " " " " " " " " "8A$$A(+A(ctjtj5tj}|dt |ddddS#1swxYwYdSrz)r{r|r r}r*rjrrDs r$test_counterexample_planar_recursivez4TestLRPlanarity.test_counterexample_planar_recursives ]2/ 0 0 , , A JJqMMM ( + + +  , , , , , , , , , , , , , , , , , ,rcd}tj|}tj|\}}|dusJ|dd||d|ddd|ddd||d|d||ddS) N)rrrrr)rrrqruTrrr(rr)rur1)r r*r rUradd_half_edge_ccwadd_half_edge_cwremove_edges_from)r,edgesr certPs r'test_edge_removal_from_planar_embeddingz7TestLRPlanarity.test_edge_removal_from_planar_embeddingsH HUOO$Q''at|||| a d+++ Aq!$$$ 1a### d+++ ,--- d+++++r graph_typec|gd}tj|\}}|sJ|}t|tjsJt d|dDsJdS)N)rrrrrrrqc3*K|]\}}}|ikVdSN).0_ds r zLTestLRPlanarity.test_graph_planar_embedding_to_undirected..-s,>>1aAG>>>>>>rTdata)r r to_undirected isinstancer*allr)r,rr rrUs r)test_graph_planar_embedding_to_undirectedz9TestLRPlanarity.test_graph_planar_embedding_to_undirected&s JGGG H H)!,, 1 OO  !RX&&&&&>>1777+=+=>>>???????rzreciprocal, as_view))TT)TF)FTctjgd}tj|\}}|sJtjt d5|||ddddS#1swxYwYdS)N)rrrrrqz%is not supported for PlanarEmbedding.match) reciprocalas_view)r r*r r{r| ValueErrorr)r,rrr rrs r6test_planar_embedding_to_undirected_invalid_parameterszFTestLRPlanarity.test_planar_embedding_to_undirected_invalid_parameters/s H=== > >)!,, 1 ]:-T U U U D D OOz7O C C C D D D D D D D D D D D D D D D D D Ds A22A69A6r)$__name__ __module__ __qualname____doc__ staticmethodrr.r3r7r;r?rErJrOrSrVr^rarcrfrhrkrsrvrxr~rrr{mark parametrizer r*r`rrrrrrr s "0"0"0\"0H666 666$MMM@@@666------777 ,,, ,,, , , ,D,,,--- ,,,--- ,,, ,,,"---0---*""",,, , , , [\BHbm+DEE@@FE@ [KKKDDDDDrrct|tjstjd|t |jt |jks Jdt }|jD]Z}|d|dkrF||d|df||d|df[|t |jks JddS)a;Raises an exception if the combinatorial embedding is not correct Parameters ---------- G : NetworkX graph embedding : a dict mapping nodes to a list of edges This specifies the ordering of the outgoing edges from a node for a combinatorial embedding Notes ----- Checks the following things: - The type of the embedding is correct - The nodes and edges match the original graph - Every half edge has its matching opposite half edge - No intersections of edges (checked by Euler's formula) z-Bad embedding. Not of type nx.PlanarEmbeddingz4Bad embedding. Nodes don't match the original graph.rrz4Bad embedding. Edges don't match the original graph.N) rr PlanarEmbeddingr}check_structuresetnodesradd)r embeddingg_edgesedges rr r <s & i!3 4 4T"#RSSS qw<<3y// / / /> 0 / / eeG,, 7d1g   KKa$q'* + + + KKa$q'* + + + c)/** * * *> + * * * *rctj|}|D].}|||r|||/t |}t |dkr|}||vr,|j|}|dkrt||}t|}t|}| || || || ||t |dkt |dkr=tj tjd|stjddSt |dkr>tj tjdd|stjddStjd)zRaises an exception if the counterexample is wrong. Parameters ---------- G : NetworkX graph subdivision_nodes : set A set of nodes inducing a subgraph as a counterexample rrr$zBad counter example.r rN)r r*has_edgerUlistlenpopdegreeiternextappend remove_noderH is_isomorphicr:r}r5)r sub_graphucontract contract_noder neighborsvs rr r fs##I((   a # # (  ! !!Q ' ' 'IH h--!     ) ) !-0 Q;;Y}566IYAYA OOA    OOA     ! !- 0 0 0   q! $ $ $# h--!  ( 9~~ 1! 4 4i@@ ?&'=>> > ? ? Y1   ;Aq A A9MM ?&'=>> > ? ?"#9:::rcleZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zed ZdS)TestPlanarEmbeddingClassctj}|ddtjtjd5|ddddddn #1swxYwYtjtjd5|ddd dddn #1swxYwYtjtjd 5|dddd dddn #1swxYwYtjtjd 5|dddddn #1swxYwY|ddd|ddd t |d dddddfddddd fddddd fgksJdS)Nrrz!Invalid clockwise reference node.rrrcwz(Invalid counterclockwise reference node.ccwz$Only one of cw/ccw can be specified.rrzNNode already has out-half-edge\(s\), either cw or ccw reference node required.Tr)rr)r r add_half_edger{r|r}sortedrr,rs rtest_add_half_edgez+TestPlanarEmbeddingClass.test_add_half_edges&(( 1%%% ]  (K    0 0  # #AqQ # / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0]  (R    1 1  # #Aqa # 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1]  (N    7 7  # #AqQA # 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7]  6    * *  # #Aq ) ) ) * * * * * * * * * * * * * * * 1+++1!,,,ioo4o0011 1A&& ' !A&& ' !A&& '6       sH A//A36A3B??CC*DDD;EE"%E"c|d}|}gddgdgdgd}||ksJdS)Nr)rrrr)rrrr)get_star_embeddingget_data)r,rrdata_cmps r test_get_dataz&TestPlanarEmbeddingClass.test_get_datasX++A.. !!## yyaSaSaS99xrc tj}|gdgdddggdgdddggdd |d |tj}|ddggdddggd ddgddggdd tj||sJdS) N)rr$r")rrrr$rr)rr r$r)r"rrrr")r rr$)rrrrr$r r"))r$rr#)rr r)r rset_datarrutils graphs_equal)r,rembedding_expecteds rtest_edge_removalz*TestPlanarEmbeddingClass.test_edge_removals&(( 99<<q6<<<<q699    ##$4555!!###/11##q6<<q699q6q699   x$$Y0BCCCCCCCrctjdiidiid}tjtj5|ddddS#1swxYwYdS)Nrrrr rr{r|r}rrs rtest_missing_edge_orientationz6TestPlanarEmbeddingClass.test_missing_edge_orientations&Ar72w'?'?@@ ]2/ 0 0 ( (  % % ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sAA #A ctjddddiddddidiidiid}tjtj5|ddddS#1swxYwYdS)Nrrrr)rrrrrrs rtest_invalid_edge_orientationz6TestPlanarEmbeddingClass.test_invalid_edge_orientations&a**+a**+r7r7     ]2/ 0 0 ( (  % % ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sA((A,/A,ctj}|ddtjtj5|ddddS#1swxYwYdSNrr)r rrr{r|r}rrs rtest_missing_half_edgez/TestPlanarEmbeddingClass.test_missing_half_edges&(( 1%%% ]2/ 0 0 ( (  % % ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sA**A.1A.cNtj}tdD]6}d}tdD]"}||kr|||||}#7t jtj5|ddddS#1swxYwYdS)Nr$r)r rrGrr{r|r}r)r,rrIrefjs r!test_not_fulfilling_euler_formulaz:TestPlanarEmbeddingClass.test_not_fulfilling_euler_formulas&(( q  AC1XX  66++AqS+999C ]2/ 0 0 ( (  % % ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s8BB!Bctj}tjtjd5|dddddddS#1swxYwYdS)NzInvalid reference node.rrrrr)r rr{r|r}rrs rtest_missing_referencez/TestPlanarEmbeddingClass.test_missing_references&(( ]2/7P Q Q Q 1 1  # #Aqa # 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1sAA!AcXtj}|dddSr)r rconnect_componentsrs rtest_connect_componentsz0TestPlanarEmbeddingClass.test_connect_componentss,&(( $$Q*****rctj}|dd|dd|dd}|ddgksJdSr)r rr traverse_face)r,rfaces rtest_successful_face_traversalz7TestPlanarEmbeddingClass.test_successful_face_traversal sf&(( 1%%%1%%%&&q!,,1v~~~~~~rctjddddiddddid}tjtj5|ddddddS#1swxYwYdS)Nrrrrr)r rr{r|r}rrs r test_unsuccessful_face_traversalz9TestPlanarEmbeddingClass.test_unsuccessful_face_traversals&1Q''(a!1D1D-E F F  ]2/ 0 0 * *  # #Aq ) ) ) * * * * * * * * * * * * * * * * * *sA$$A(+A(c2tj}|d|dgt jt 5|dddddn #1swxYwYt jt 5|ddgdddn #1swxYwYt jt 5| ddgddddS#1swxYwYdS) N*)rrrqrX)rri^)rr}) r rrjadd_nodes_fromr{r|NotImplementedErrorrHrCadd_weighted_edges_fromrs rtest_forbidden_methodsz/TestPlanarEmbeddingClass.test_forbidden_methodss&(( 2  (,,, ]. / / % %   q! $ $ $ % % % % % % % % % % % % % % % ]. / / 7 7  $ $ff%5 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 ]. / / J J  - -{K.H I I I J J J J J J J J J J J J J J J J J Js6A;;A?A?CC C'D  DDctj}d}td|D]2}|d|||}||d3|S)Nrrr)r rrGr)nrrrIs rrz+TestPlanarEmbeddingClass.get_star_embedding$sk&(( q! * *A  # #AqS # 1 1 1C  # #Aq ) ) ) )rN)rrrrrrrrrrrrrrrrrrrrrrs   @   DDD<((( ( ( (((( ( ( (111 +++*** J J J\rr) r{networkxr networkx.algorithms.planarityrrrrr r rrrrrs nDnDnDnDnDnDnDnDb '''T0;0;0;fSSSSSSSSSSr