3 Md<@sFdZddlZddlmZddlZejdZejdZGdddZ dS)z Unit tests for layout functions.N) almost_equalnumpyZscipyc@seZdZeddZeddZddZddZd d Z d d Z d dZ ddZ ddZ ddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Z d9S): TestLayoutcCs8tjdd|_tj|_tj|jdtjdd|_dS)NZabcdef)nxZ grid_2d_graphGiGraphGsZadd_pathbigG)clsr T/var/www/html/virt/lib/python3.6/site-packages/networkx/drawing/tests/test_layout.py setup_class s zTestLayout.setup_classcCsLg}d}x>|D]6}|dk r<|||}|jtj||d||}qW|S)Ng?)appendrdot)Z positions distancesZprev_valkdiffr r rcollect_node_distancess   z!TestLayout.collect_node_distancescCs\tjd}tjttj|dgddd d}tjttj|ddg|dtj|ddg|ddS) Nr)fixed)r)rposr)rr)rr)r path_graphpytestraises ValueError spring_layout)selfGrr r rtest_spring_fixed_without_poss   z(TestLayout.test_spring_fixed_without_poscslddltj}|jd d d d gddi}dg}tj|||d}tfdd|jD}| shtd dS)Nrrr)rrc3s"|]}|D]}j|Vq qdS)N)isnan).0Zcoordsc)mathr r /sz2TestLayout.test_spring_init_pos..zvalues should not be nan)rr)rr)rr)rr#)r$r$)r(rr Zadd_edges_fromfruchterman_reingold_layoutanyvaluesAssertionError)r r!Zinit_posZ fixed_posrZhas_nanr )r(rtest_spring_init_pos%szTestLayout.test_spring_init_poscCsxg}tj|tj|tj|tj|tj|tj|tj|tj||tj |tj |tj |dS)N) r random_layoutcircular_layout planar_layoutrr*spectral_layout shell_layoutbipartite_layout spiral_layoutmultipartite_layoutkamada_kawai_layout)r r!r r rtest_smoke_empty_graph2s          z!TestLayout.test_smoke_empty_graphcCs|j}tj|tj|tj|tj|tj|tj|jtj|tj|j tj|jtj|jj tj |tj |tj |tj |ddtj |dddS)Nr)dimr#) rrr/r0r1rr*r r2Z to_directedr3r5r7)r r!r r rtest_smoke_int@s            zTestLayout.test_smoke_intcCs|j}tj|tj|tj|tj|tj|tj|tj|tj |tj |tj |ddtj |dddS)Nr)r9r#) r rr/r0r1rr*r2r3r5r7)r r!r r rtest_smoke_stringRs         zTestLayout.test_smoke_stringcCsxtj|}||}||}tjt|j}|jd|jd}|d|kjsTt||kjsdt||kjsttdS)Nrr)rarraylistr,maxminallr-)r rscalecenterlowhivposlengthr r rcheck_scale_and_center`s z!TestLayout.check_scale_and_centercCs|j}d }tjd}|jd|tj||ddd d|tj|d |dd |d|tj|d |dd |d|tj|d |dd |d|tj|d |dd |d|tj |d |dd |d|tj |d |dd |dd}|tj |d d |d d |ddS)Nrr )rBg?@@)rArBrr#)r9rArB)rr)rIrJ)rr#r) rGrcomplete_graphadd_noder/rr2r0r3r5r7)r scr'r!r r rtest_scale_and_center_argjs  z$TestLayout.test_scale_and_center_argcCs tjd}tjtjtj|dS)NrH)rrKrrZNetworkXExceptionr1)r r!r r r#test_planar_layout_non_planar_input{s z.TestLayout.test_planar_layout_non_planar_inputcCs4tj}|jddgddgddgdtj|dS)Nrrr)rrr)rZPlanarEmbeddingset_datar1)r Z embeddingr r r(test_smoke_planar_layout_embedding_inputsz3TestLayout.test_smoke_planar_layout_embedding_inputcCs|j}d}tjd}|jd|tj|dd d|tj|d|d|tj|d|d|tj|d|d|tj|d|d|tj |d|d|tj |d|dd }|tj |ddd|ddS) NrrH?)rArBrr#)r9)rr)rRrR)rrr) rGrrKrLr/rr2r0r3r5r7)r rMr'r!r r rtest_default_scale_and_centers  z(TestLayout.test_default_scale_and_centercCsrtjd}tjttj|ddtjttj|ddtjttj|ddtjttj|ddtjttj|dddS)Nrr)r9r#)rrrrrr0r3r1)r r!r r r(test_circular_planar_and_shell_dim_errors  z3TestLayout.test_circular_planar_and_shell_dim_errorcCshtj|j}tjjj|}|jdks(ttjjj|dd}|jdksHttjjj|}|jdksdtdS)Nrr#)r9)rUr)rUr#)rUr) rZto_numpy_arrayr drawinglayoutZ_fruchterman_reingoldshaper-_sparse_fruchterman_reingold)r Arr r rtest_adjacency_interface_numpys z)TestLayout.test_adjacency_interface_numpycCsltj|jdd}tjjj|}|jdks,ttjjj|}|jdksHttjjj|dd}|jd kshtdS) Nd)ZdtyperUrr#)r9)rUr)rUr)rUr#) rZto_scipy_sparse_matrixr rVrWrYrXr-Z_sparse_spectral)r rZrr r rtest_adjacency_interface_scipysz)TestLayout.test_adjacency_interface_scipycCstjd}tj|}|dj s&ttjd}tj|dgddgdgg}|dj s\t|djslttjj|ddksttj|dgddgdggdd}tjj|ddkstdS)Nrrrrr#)rotate)rrr3r+r-rlinalgnorm)r r!rEr r rtest_single_nodess   zTestLayout.test_single_nodescCs tj|j}tj|j|d}dS)N)r)rr0rr*)r rnposr r r+test_smoke_initial_pos_fruchterman_reingolds z6TestLayout.test_smoke_initial_pos_fruchterman_reingoldcCstj|j}tj|j|dgd}t|dt|dks.rrm)rnrArBc3s&|]}d|dkVqdS)rNr )r&rv)rwrEr rr)1sro)rn)rrrtrru)rr) rZcomplete_multipartite_graphr6rpr-r@rerrr)r sizesr!nendr )rwrErtest_multipartite_layouts"   &  & z#TestLayout.test_multipartite_layoutcCs~tjjj}tjddg}dtjddgddgg}||t|ddd\}}t|dsVtt|ddshtt|dd sztdS)Ng@g@rg?g@g333333?r) meanweightr9r#rg?g?g?g?g)rrVrW_kamada_kawai_costfnrr<rr-)r costfnrinvdistcostgradr r rtest_kamada_kawai_costfn_1d6s z&TestLayout.test_kamada_kawai_costfn_1dc Cstjjj}||jt|||d\}}d|tjtj|ddd}xht|jdD]V} xPt| d|jdD]8} tj j || || } || || | dd7}qnWqTWt ||st d} xt|jdD]} xt|jdD]}| |jd|}|j }||| 7<||t|||jddd}||d| 8<||t|||jddd}t ||||d| d d st qWqWdS) N)r|r9g?r)rfrrg?g-C6?r)Zplaces)rrVrWr}ZravelrsumrerXr_r`rr-flatten)r rrmeanwtr9r~rrZ expected_costrvjrZdxnddmidxZpos0ZcplusZcminusr r rcheck_kamada_kawai_costfnBs,  $z$TestLayout.check_kamada_kawai_costfncCsdtjdddgdddgdddgg}d}tjddgd dgd d gg}|j|||d tjdddgdddgdddgg}|j|||ddS)Nrg?g@g333333?g?g333333?g333333?g?g @g@gffffff@g@rg?g333333!@gffffff!@ g?gffffff@g333333"@g333333 @g?r#g g333333ӿgffffff!iggffffffg333333 )rr<r)r rrrr r rtest_kamada_kawai_costfnas &"z#TestLayout.test_kamada_kawai_costfnc Cs|j}tj|dd}tj|dd}|j|}|j|}t|t|ksJttj|dd}|j|}x6tdt|dD] }t||||ddsvtqvWdS)Ngffffff?) resolutiong(\?T)Z equidistantrr) r rr5rrr-rerpr) r r!Z pos_standardZ pos_tighterrZdistances_tighterZpos_equidistantZdistances_equidistantr\r r rtest_spiral_layoutos   zTestLayout.test_spiral_layoutcCstj}tj|dd}tj|iks(ttjd}ddd}tj|}tjj}|ddt|j Dd kspttjd }dddd }tj|}|dddd ksttj|dd}|dddd kstdS)Nr)rBr?)rrcSsg|] }t|qSr )r)r&xr r r sz7TestLayout.test_rescale_layout_dict..gư>r#r?)rrr)rA)rr)rr)rr)rr)rr)rrr)rr)rr)rrr)rr)rr)rr) rrkr/Zrescale_layout_dictr-rr_r`zipr,)r r!rEZs_vposr`r r rtest_rescale_layout_dicts   "   z#TestLayout.test_rescale_layout_dictN)!__name__ __module__ __qualname__ classmethodr staticmethodrr"r.r8r:r;rGrNrOrQrSrTr[r]rarcrgrhrjrlrsr{rrrrrr r r rr s8         r) __doc__ZnetworkxrZnetworkx.testingrrZ importorskiprZtest_smoke_empty_graphscipyrr r r rs