a
    h"                     @   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J d S )"N   Zcreate_usingc                 S   s   i | ]}|||fqS  r   ).0nr   r   O/var/www/auris/lib/python3.9/site-packages/networkx/drawing/tests/test_latex.py
<dictcomp>       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edgesnodesto_latex)Gr   
output_texexpected_texr   r   r   test_tikz_attributes   s<    r   c                  C   s   t d} t d}t d}t d}g d}g d}t j| |||gd||d}d|v s^J d|v sjJ |d	dks|J |d
dksJ |ddksJ d S )Nr      )zPath on 4 nodeszComplete graph on 4 nodeszPath on 8 nodeszComplete graph on 8 nodes)Z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   count)ZH1ZH2ZH3ZH4ZcaptionslabelsZ
latex_coder   r   r   test_basic_multiple_graphsA   s"    




r&   c            
      C   s   d} g d}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sJ 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   Zrc_node_colorZgp_node_colorHr"   r#   r   r   r   r   test_basic_tikz[   sF    J
r/   c                 C   s   t d}| |dd dddd}tt j | || W d    n1 sN0    Y  d|d	< tt j | || W d    n1 s0    Y  d
|d	< tt j | || W d    n1 s0    Y  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    
(((r5   c                 C   s  t d}ddd}dddd}||||g}||||g}| || tt j | || W d    n1 sn0    Y  | || tt j" | |||||g W d    n1 s0    Y  tt j" | |||||g W d    n1 s0    Y  tt j" | ||ddgd W d    n1 s40    Y  tt j" | ||ddgd W d    n1 st0    Y  | ||dgd	 d
gd	 d d S )Nr   r   r   r0   r1   hi)r"   )r#   r   Zlbl)r"   r#   r2   )r   r   Zpos_badZpos_OKZfourGZfourposr   r   r   test_exception_multiple_graphs   s$    


(
0022r7   c                  C   sV   t jdt jd} | dd tt j t |  W d    n1 sH0    Y  d S )Nr   r   r   r   )r   r   Z
MultiGraphr   r3   r4   ZNetworkXNotImplementedr   )r   r   r   r   test_exception_multigraph   s    r8   )
r3   Znetworkxr   r   r&   r/   r   r5   r7   r8   r   r   r   r   <module>   s   ;  