U md"@sPddlZddlZddZddZddZejfdd Zejfd d Zd d Z dS)Nc Cs tjdtjd}dd|D}|ddd|jdd<d |jdd <d |jdd <d |jdd <d|jdd <d|jdd<d|jdd<d|jdd <d|jdd<d|jdd <d|jdd <d|jdd<d|jdd <d|jdd <tj||ddd d ddd d }d!}||kstdS)"NZ create_usingcSsi|]}|||fqSr).0nrrZ/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/networkx/drawing/tests/test_latex.py sz(test_tikz_attributes..rZLoop)rrlabelZmidwayZ label_optionsbluestylezline width=3,drawzcircle,draw,blue!50ZStopz1st Steprr znear endz3rd Step)r rz near startzbend left,greenZ2ndr r zpos=0.5z$>->,bend right,line width=3,green!90Fz [scale=3])pos as_document tikz_options node_optionsZ edge_optionsZ node_labelZ edge_labelZedge_label_optionsa(\begin{figure} \begin{tikzpicture}[scale=3] \draw (0, 0) node[blue] (0){0} (1, 1) node[line width=3,draw] (1){1} (2, 2) node[circle,draw,blue!50] (2){2} (3, 3) node (3){Stop}; \begin{scope}[->] \draw (0) to node[near end] {1st Step} (1); \draw[loop,] (0) to node[midway] {Loop} (0); \draw[>->,bend right,line width=3,green!90] (1) to node[pos=0.5] {2nd} (2); \draw[bend left,green] (2) to node[near start] {3rd Step} (3); \end{scope} \end{tikzpicture} \end{figure})nx path_graphDiGraphadd_edgeedgesZnodesto_latexAssertionError)Gr output_tex expected_texrrrtest_tikz_attributess<  rcCstd}td}td}td}ddddg}ddd d g}tj||||gd ||d }d |ksftd|ksrt|ddkst|ddkst|ddkstdS)NrzPath on 4 nodeszComplete graph on 4 nodeszPath on 8 nodeszComplete graph on 8 nodesZfig2aZfig2bZfig2cZfig2dr )n_rows sub_captions sub_labelszbegin{document}z begin{figure}zbegin{subfigure}Z tikzpicturez[-])rrZcomplete_graphrrcount)ZH1ZH2ZH3ZH4ZcaptionslabelsZ latex_coderrrtest_basic_multiple_graphsAs*        r&c Csd}dddddddd d d d d ddddg}t}|td||ddddddddd}ddddd}ddd d d}|}t||d!t||d!d"d#g}d$d%g}tj||g||gd&d'd'd!||d(d)d*d+ } | |kstdS),NaF\documentclass{report} \usepackage{tikz} \usepackage{subcaption} \begin{document} \begin{figure} \begin{subfigure}{0.5\textwidth} \begin{tikzpicture}[scale=2] \draw[gray!90] (0.749, 0.702) node[red!90] (0){0} (1.0, -0.014) node[red!90] (1){1} (-0.777, -0.705) node (2){2} (-0.984, 0.042) node (3){3} (-0.028, 0.375) node[cyan!90] (4){4} (-0.412, 0.888) node (5){5} (0.448, -0.856) node (6){6} (0.003, -0.431) node[cyan!90] (7){7}; \begin{scope}[->,gray!90] \draw (0) to (4); \draw (0) to (5); \draw (0) to (6); \draw (0) to (7); \draw (1) to (4); \draw (1) to (5); \draw (1) to (6); \draw (1) to (7); \draw (2) to (4); \draw (2) to (5); \draw (2) to (6); \draw (2) to (7); \draw (3) to (4); \draw (3) to (5); \draw (3) to (6); \draw (3) to (7); \end{scope} \end{tikzpicture} \caption{My tikz number 1 of 2}\label{tikz_1_2} \end{subfigure} \begin{subfigure}{0.5\textwidth} \begin{tikzpicture}[scale=2] \draw[gray!90] (0.749, 0.702) node[green!90] (0){0} (1.0, -0.014) node[green!90] (1){1} (-0.777, -0.705) node (2){2} (-0.984, 0.042) node (3){3} (-0.028, 0.375) node[purple!90] (4){4} (-0.412, 0.888) node (5){5} (0.448, -0.856) node (6){6} (0.003, -0.431) node[purple!90] (7){7}; \begin{scope}[->,gray!90] \draw (0) to (4); \draw (0) to (5); \draw (0) to (6); \draw (0) to (7); \draw (1) to (4); \draw (1) to (5); \draw (1) to (6); \draw (1) to (7); \draw (2) to (4); \draw (2) to (5); \draw (2) to (6); \draw (2) to (7); \draw (3) to (4); \draw (3) to (5); \draw (3) to (6); \draw (3) to (7); \end{scope} \end{tikzpicture} \caption{My tikz number 2 of 2}\label{tikz_2_2} \end{subfigure} \caption{A graph generated with python and latex.} \end{figure} \end{document})rr)r)r)r)r r)r r')r r()r r))r r)r r')r r()r r))rr)rr')rr()rr)r )g`5 ?g'y?)g?gMR )g/κg ƒ)gw!g^c?)g͛YgI1?)gDQVڿg]]Pj?)gGgǨ?g'e)g@j?g+?Cۿ)rr r rrr'r(r)zred!90zcyan!90)rr rr)zgreen!90z purple!90colorzMy tikz number 1 of 2zMy tikz number 2 of 2Ztikz_1_2Ztikz_2_2z [scale=2]zgray!90z(A graph generated with python and latex.r T) rZdefault_node_optionsZdefault_edge_optionsrr"r#captionr!r) rrZadd_nodes_fromrangeZadd_edges_fromcopyZset_node_attributesrr) rrrrZ rc_node_colorZ gp_node_colorHr"r#rrrrtest_basic_tikz[sfK  r/c Cstd}||dddddd}ttj|||W5QRXd|d <ttj|||W5QRXd |d <ttj|||W5QRXd |d <|||dS) Nrr)rrrr r rr r )r r rrr )rr rrpytestraisesZ NetworkXError)rrrrrrtest_exception_pos_single_graphs   r5c Cs6td}ddd}dddd}||||g}||||g}|||ttj|||W5QRX|||ttj||||||gW5QRXttj||||||gW5QRXttj|||ddgdW5QRXttj|||ddgdW5QRX|||dgd d gd d dS) Nrrrr0r1hi)r")r#rZlbl)r"r#r2)rrZpos_badZpos_OKZfourGZfourposrrrtest_exception_multiple_graphss$       r7c CsBtjdtjd}|ddttjt|W5QRXdS)Nrrr r )rrZ MultiGraphrr3r4ZNetworkXNotImplementedr)rrrrtest_exception_multigraph s r8) r3Znetworkxrrr&r/rr5r7r8rrrrs;