U
    9vf"                     @   sP   d dl Z d dlZdd Zdd Zdd Zejfdd	Zejfd
dZdd Z	dS )    Nc                  C   s   t jdt jd} dd | D }| dd d| jd d< d	| jd d
< d| jd d< d| jd d< d| jd d< d| jd d< d| jd d< d| jd d
< d| jd d< d| jd d
< d| jd d< d| jd d< d| jd d
< d| jd d< t j| |ddddddd
d 	}d!}||kstd S )"N   Zcreate_usingc                 S   s   i | ]}|||fqS  r   ).0nr   r   E/tmp/pip-unpacked-wheel-_lngutwb/networkx/drawing/tests/test_latex.py
<dictcomp>   s      z(test_tikz_attributes.<locals>.<dictcomp>r   ZLoop)r   r   labelZmidwayZlabel_optionsZbluestylezline width=3,draw   zcircle,draw,blue!50   ZStop   z1st Stepr   r   znear endz3rd Step)r   r   z
near startzbend left,greenZ2ndr   r   zpos=0.5z$>->,bend right,line width=3,green!90Fz	[scale=3])posas_documenttikz_optionsnode_optionsZedge_optionsZ
node_labelZ
edge_labelZedge_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_graphDiGraphadd_edgeedgesZnodesto_latexAssertionError)Gr   
output_texexpected_texr   r   r   test_tikz_attributes   s<    r   c                  C   s   t d} t d}t d}t d}ddddg}ddd	d
g}t j| |||gd||d}d|ksftd|ksrt|ddkst|ddkst|ddkstd S )Nr      zPath on 4 nodeszComplete graph on 4 nodeszPath on 8 nodeszComplete graph on 8 nodesZfig2aZfig2bZfig2cZfig2dr   )n_rowssub_captions
sub_labelszbegin{document}zbegin{figure}zbegin{subfigure}Ztikzpicturez[-])r   r   Zcomplete_graphr   r   count)ZH1ZH2ZH3ZH4ZcaptionslabelsZ
latex_coder   r   r   test_basic_multiple_graphsA   s*    




r%   c            
      C   s   d} dddddddd	d
dddddddg}t  }|td || ddddddddd}ddddd}ddd d d}| }t ||d! t ||d! d"d#g}d$d%g}t j||g||gd&d'd'd!||d(d)d*d+}	|	| kstd S ),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})r   r   )r      )r      )r      )r   r   )r   r&   )r   r'   )r   r(   )r   r   )r   r&   )r   r'   )r   r(   )r   r   )r   r&   )r   r'   )r   r(   r   )g`5?g'y?)g      ?gMR )g/κg	ƒ)gw!g^c?)g͛YgI1?)gDQVڿg]]Pj?)gGgǨ?g'e)g@j?g+?Cۿ)r   r   r   r   r   r&   r'   r(   zred!90zcyan!90)r   r   r   r(   zgreen!90z	purple!90colorzMy tikz number 1 of 2zMy tikz number 2 of 2Ztikz_1_2Ztikz_2_2z	[scale=2]zgray!90z(A graph generated with python and latex.r   T)	r   Zdefault_node_optionsZdefault_edge_optionsr   r!   r"   captionr    r   )	r   r   Zadd_nodes_fromrangeZadd_edges_fromcopyZset_node_attributesr   r   )
r   r   r   r   Zrc_node_colorZgp_node_colorHr!   r"   r   r   r   r   test_basic_tikz[   sf    K
r.   c              	   C   s   t d}| |dd dddd}tt j | || W 5 Q R X d|d	< tt j | || W 5 Q R X d
|d	< tt j | || W 5 Q R X d|d	< | || d S )Nr   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   )r   r   r   r   pytestraisesZNetworkXError)r   r   r   r   r   r   test_exception_pos_single_graph   s    
r4   c              	   C   s6  t d}ddd}dddd}||||g}||||g}| || tt j | || W 5 Q R X | || tt j | |||||g W 5 Q R X tt j | |||||g W 5 Q R X tt j | ||ddgd W 5 Q R X tt j | ||ddgd W 5 Q R X | ||dgd	 d
gd	 d d S )Nr   r   r   r/   r0   hi)r!   )r"   r   Zlbl)r!   r"   r1   )r   r   Zpos_badZpos_OKZfourGZfourposr   r   r   test_exception_multiple_graphs   s$    



r6   c               	   C   sB   t jdt jd} | dd tt j t |  W 5 Q R X d S )Nr   r   r   r   )r   r   Z
MultiGraphr   r2   r3   ZNetworkXNotImplementedr   )r   r   r   r   test_exception_multigraph   s    r7   )
r2   Znetworkxr   r   r%   r.   r   r4   r6   r7   r   r   r   r   <module>   s   ;  