o
    ]ZŽhØ  ã                   @   s@  d dl Z d dlmZ d dlmZ d dlZd dlZdd„ Zdd„ Z	dd	„ Z
d
d„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd d!„ Zd"d#„ Zd$d%„ Zd&d'„ Zd(d)„ Zd*d+„ Zd,d-„ Zd.d/„ Zd0d1„ Zd2d3„ Zd4d5„ Z d6d7„ Z!d8d9„ Z"d:d;„ Z#d<d=„ Z$ej% &d>e'd?d@„ dAD ƒƒ¡dBdC„ ƒZ(dS )Dé    N)Úproduct)Údedentc                  C   s˜   t jddt jd} | jD ]}dttdƒ| ƒ | j| d< qtdƒ ¡ }tdƒ ¡ }t j| dd	}d
 	|¡|ks:J ‚t j| dd	}d
 	|¡|ksJJ ‚d S )Né   ©ÚrÚhÚcreate_usingZnode_ÚaÚlabeluÁ   
        â•™â”€â”€ 0
            â”œâ”€â•¼ 1
            â”‚   â”œâ”€â•¼ 3
            â”‚   â””â”€â•¼ 4
            â””â”€â•¼ 2
                â”œâ”€â•¼ 5
                â””â”€â•¼ 6
        uä   
        â•™â”€â”€ node_a
            â”œâ”€â•¼ node_b
            â”‚   â”œâ”€â•¼ node_d
            â”‚   â””â”€â•¼ node_e
            â””â”€â•¼ node_c
                â”œâ”€â•¼ node_f
                â””â”€â•¼ node_g
        F)Úwith_labelsÚ
T)
ÚnxÚbalanced_treeÚDiGraphÚnodesÚchrÚordr   ÚstripÚgenerate_network_textÚjoin)ÚgraphÚnodeZnode_targetZlabel_targetÚret© r   úQ/var/www/auris/lib/python3.10/site-packages/networkx/readwrite/tests/test_text.pyÚ*test_generate_network_text_forest_directed
   s"   
 ÿ
öÿ
ör   c                  C   sd   dd„ } | t  ¡ ƒdksJ ‚| t  ¡ ƒdksJ ‚| t  ¡ dddks$J ‚| t  ¡ dddks0J ‚d S )Nc                 [   s(   g }t j| |jfddi|¤Ž d |¡S )NÚendÚ r   )r   Úwrite_network_textÚappendr   )ÚgÚkwZprintbufr   r   r   Ú
_graph_str2   s   
z7test_write_network_text_empty_graph.<locals>._graph_stru   â•™T)Ú
ascii_onlyú+)r   r   ÚGraph)r"   r   r   r   Ú#test_write_network_text_empty_graph1   s
   r&   c                  C   st   t  ¡ } |  g d¢¡ |  dd¡ g }|j}t j| |dd t j| |ddd d |¡}td	ƒ ¡ }||ks8J ‚d S )
N©é   r   é   é   r   r*   r   ©Úpathr   T©r,   r#   r   r   u›   
        â•Ÿâ”€â”€ 1
        â•Ÿâ”€â”€ 2
        â•Ž   â””â”€â•¼ 4
        â•™â”€â”€ 3
        +-- 1
        +-- 2
        :   L-> 4
        +-- 3
        )	r   r   Úadd_nodes_fromÚadd_edger   r   r   r   r   )r    ÚlinesÚwriteÚtextÚtargetr   r   r   Ú+test_write_network_text_within_forest_glyph=   s   
ÿõr4   c                  C   sâ   t jddt jd} t jddt jd}t  | |g¡}d t  |¡¡}tdƒ ¡ }||ks-J ‚t jddt jd}t  | ||g¡}d t j|g d¢d¡}tdƒ ¡ }||ksVJ ‚d t j|g d¢dd	¡}td
ƒ ¡ }||ksoJ ‚d S )Nr   r   r   u‰  
        â•Ÿâ”€â”€ 0
        â•Ž   â”œâ”€â•¼ 1
        â•Ž   â”‚   â”œâ”€â•¼ 3
        â•Ž   â”‚   â””â”€â•¼ 4
        â•Ž   â””â”€â•¼ 2
        â•Ž       â”œâ”€â•¼ 5
        â•Ž       â””â”€â•¼ 6
        â•™â”€â”€ 7
            â”œâ”€â•¼ 8
            â”‚   â”œâ”€â•¼ 10
            â”‚   â””â”€â•¼ 11
            â””â”€â•¼ 9
                â”œâ”€â•¼ 12
                â””â”€â•¼ 13
        )r   é   é   ©ÚsourcesuT  
        â•Ÿâ”€â”€ 0
        â•Ž   â”œâ”€â•¼ 1
        â•Ž   â”‚   â”œâ”€â•¼ 3
        â•Ž   â”‚   â””â”€â•¼ 4
        â•Ž   â””â”€â•¼ 2
        â•Ž       â”œâ”€â•¼ 5
        â•Ž       â””â”€â•¼ 6
        â•Ÿâ”€â”€ 14
        â•Ž   â”œâ”€â•¼ 15
        â•Ž   â”‚   â”œâ”€â•¼ 17
        â•Ž   â”‚   â””â”€â•¼ 18
        â•Ž   â””â”€â•¼ 16
        â•Ž       â”œâ”€â•¼ 19
        â•Ž       â””â”€â•¼ 20
        â•™â”€â”€ 7
            â”œâ”€â•¼ 8
            â”‚   â”œâ”€â•¼ 10
            â”‚   â””â”€â•¼ 11
            â””â”€â•¼ 9
                â”œâ”€â•¼ 12
                â””â”€â•¼ 13
        T©r8   r#   a²  
        +-- 0
        :   |-> 1
        :   |   |-> 3
        :   |   L-> 4
        :   L-> 2
        :       |-> 5
        :       L-> 6
        +-- 14
        :   |-> 15
        :   |   |-> 17
        :   |   L-> 18
        :   L-> 16
        :       |-> 19
        :       L-> 20
        +-- 7
            |-> 8
            |   |-> 10
            |   L-> 11
            L-> 9
                |-> 12
                L-> 13
        )r   r   r   Údisjoint_union_allr   r   r   r   )Útree1Útree2Úforestr   r3   Ztree3r   r   r   Ú.test_generate_network_text_directed_multi_treeU   s8   ÿïÿèÿÿèr>   c                     s²   t jddt jd‰ t jddt jd} t  | ‡ fdd„| jD ƒ¡} t  ˆ | ¡}d t j|ddgd¡}td	ƒ 	¡ }||ks>J ‚d t j|ddgd
d¡}tdƒ 	¡ }||ksWJ ‚d S )Nr   r   c                    s   i | ]	}||t ˆ ƒ “qS r   )Úlen©Ú.0Ún©r;   r   r   Ú
<dictcomp>±   s    zDtest_generate_network_text_undirected_multi_tree.<locals>.<dictcomp>r   r   r6   r7   u‰  
        â•Ÿâ”€â”€ 0
        â•Ž   â”œâ”€â”€ 1
        â•Ž   â”‚   â”œâ”€â”€ 3
        â•Ž   â”‚   â””â”€â”€ 4
        â•Ž   â””â”€â”€ 2
        â•Ž       â”œâ”€â”€ 5
        â•Ž       â””â”€â”€ 6
        â•™â”€â”€ 7
            â”œâ”€â”€ 8
            â”‚   â”œâ”€â”€ 10
            â”‚   â””â”€â”€ 11
            â””â”€â”€ 9
                â”œâ”€â”€ 12
                â””â”€â”€ 13
        Tr9   a!  
        +-- 0
        :   |-- 1
        :   |   |-- 3
        :   |   L-- 4
        :   L-- 2
        :       |-- 5
        :       L-- 6
        +-- 7
            |-- 8
            |   |-- 10
            |   L-- 11
            L-- 9
                |-- 12
                L-- 13
        )
r   r   r%   Úrelabel_nodesr   Úunionr   r   r   r   )r<   r=   r   r3   r   rC   r   Ú0test_generate_network_text_undirected_multi_tree®   s$   ÿïÿïrG   c                  C   sr   t jddt jd} tdƒ ¡ }d t j| dgd¡}||ks J ‚tdƒ ¡ }d t j| dgd¡}||ks7J ‚d S )Nr   r   uÁ   
        â•™â”€â”€ 0
            â”œâ”€â”€ 1
            â”‚   â”œâ”€â”€ 3
            â”‚   â””â”€â”€ 4
            â””â”€â”€ 2
                â”œâ”€â”€ 5
                â””â”€â”€ 6
        r   r   r7   uÇ   
        â•™â”€â”€ 2
            â”œâ”€â”€ 0
            â”‚   â””â”€â”€ 1
            â”‚       â”œâ”€â”€ 3
            â”‚       â””â”€â”€ 4
            â”œâ”€â”€ 5
            â””â”€â”€ 6
        )r   r   r%   r   r   r   r   )r   Znode_target0r   Znode_target2r   r   r   Ú,test_generate_network_text_forest_undirectedà   s   ÿ
öÿ
örH   c               	   C   s”   t  t jddt jdt jddt jdt jddt jdg¡} tdƒ ¡ }tdƒ ¡ }d t j| | jd¡}d t  | ¡¡}||ksBJ ‚||ksHJ ‚dS )	zª
    When sources are directly specified, we won't be able to determine when we
    are in the last component, so there will always be a trailing, leftmost
    pipe.
    r   r(   r   uå   
        â•Ÿâ”€â”€ 0
        â•Ž   â”œâ”€â•¼ 1
        â•Ž   â””â”€â•¼ 2
        â•Ÿâ”€â”€ 3
        â•Ž   â””â”€â•¼ 4
        â•Ž       â””â”€â•¼ 5
        â•Ÿâ”€â”€ 6
        â•Ž   â”œâ”€â•¼ 7
        â•Ž   â””â”€â•¼ 8
        uá   
        â•Ÿâ”€â”€ 0
        â•Ž   â”œâ”€â•¼ 1
        â•Ž   â””â”€â•¼ 2
        â•Ÿâ”€â”€ 3
        â•Ž   â””â”€â•¼ 4
        â•Ž       â””â”€â•¼ 5
        â•™â”€â”€ 6
            â”œâ”€â•¼ 7
            â””â”€â•¼ 8
        r   r7   N)	r   r:   r   r   r   r   r   r   r   )r   Ztarget1Ztarget2Zgot1Zgot2r   r   r   Ú0test_generate_network_text_overspecified_sources  s(   ýÿ	ÿôÿôrI   c                  C   s¨   t  ¡ } |  g d¢¡ g }|j}|dƒ t j| |dd t| j| jƒD ]\}}|d|› d|› dƒ |  ||¡ t j| |dd q#d |¡}t	d	ƒ 
¡ }||ksRJ ‚d
S )úS
    Walk through the cases going from a disconnected to fully connected graph
    r'   ú--- initial state ---r   r+   ú--- add_edge(ú, ú) ---r   uk  
        --- initial state ---
        â•Ÿâ”€â”€ 1
        â•Ÿâ”€â”€ 2
        â•Ÿâ”€â”€ 3
        â•™â”€â”€ 4
        --- add_edge(1, 1) ---
        â•Ÿâ”€â”€ 1 â•¾ 1
        â•Ž   â””â”€â•¼  ...
        â•Ÿâ”€â”€ 2
        â•Ÿâ”€â”€ 3
        â•™â”€â”€ 4
        --- add_edge(1, 2) ---
        â•Ÿâ”€â”€ 1 â•¾ 1
        â•Ž   â”œâ”€â•¼ 2
        â•Ž   â””â”€â•¼  ...
        â•Ÿâ”€â”€ 3
        â•™â”€â”€ 4
        --- add_edge(1, 3) ---
        â•Ÿâ”€â”€ 1 â•¾ 1
        â•Ž   â”œâ”€â•¼ 2
        â•Ž   â”œâ”€â•¼ 3
        â•Ž   â””â”€â•¼  ...
        â•™â”€â”€ 4
        --- add_edge(1, 4) ---
        â•™â”€â”€ 1 â•¾ 1
            â”œâ”€â•¼ 2
            â”œâ”€â•¼ 3
            â”œâ”€â•¼ 4
            â””â”€â•¼  ...
        --- add_edge(2, 1) ---
        â•™â”€â”€ 2 â•¾ 1
            â””â”€â•¼ 1 â•¾ 1
                â”œâ”€â•¼ 3
                â”œâ”€â•¼ 4
                â””â”€â•¼  ...
        --- add_edge(2, 2) ---
        â•™â”€â”€ 1 â•¾ 1, 2
            â”œâ”€â•¼ 2 â•¾ 2
            â”‚   â””â”€â•¼  ...
            â”œâ”€â•¼ 3
            â”œâ”€â•¼ 4
            â””â”€â•¼  ...
        --- add_edge(2, 3) ---
        â•™â”€â”€ 1 â•¾ 1, 2
            â”œâ”€â•¼ 2 â•¾ 2
            â”‚   â”œâ”€â•¼ 3 â•¾ 1
            â”‚   â””â”€â•¼  ...
            â”œâ”€â•¼ 4
            â””â”€â•¼  ...
        --- add_edge(2, 4) ---
        â•™â”€â”€ 1 â•¾ 1, 2
            â”œâ”€â•¼ 2 â•¾ 2
            â”‚   â”œâ”€â•¼ 3 â•¾ 1
            â”‚   â”œâ”€â•¼ 4 â•¾ 1
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- add_edge(3, 1) ---
        â•™â”€â”€ 2 â•¾ 1, 2
            â”œâ”€â•¼ 1 â•¾ 1, 3
            â”‚   â”œâ”€â•¼ 3 â•¾ 2
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”œâ”€â•¼ 4 â•¾ 2
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- add_edge(3, 2) ---
        â•™â”€â”€ 3 â•¾ 1, 2
            â”œâ”€â•¼ 1 â•¾ 1, 2
            â”‚   â”œâ”€â•¼ 2 â•¾ 2, 3
            â”‚   â”‚   â”œâ”€â•¼ 4 â•¾ 1
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- add_edge(3, 3) ---
        â•™â”€â”€ 1 â•¾ 1, 2, 3
            â”œâ”€â•¼ 2 â•¾ 2, 3
            â”‚   â”œâ”€â•¼ 3 â•¾ 1, 3
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”œâ”€â•¼ 4 â•¾ 1
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- add_edge(3, 4) ---
        â•™â”€â”€ 1 â•¾ 1, 2, 3
            â”œâ”€â•¼ 2 â•¾ 2, 3
            â”‚   â”œâ”€â•¼ 3 â•¾ 1, 3
            â”‚   â”‚   â”œâ”€â•¼ 4 â•¾ 1, 2
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- add_edge(4, 1) ---
        â•™â”€â”€ 2 â•¾ 1, 2, 3
            â”œâ”€â•¼ 1 â•¾ 1, 3, 4
            â”‚   â”œâ”€â•¼ 3 â•¾ 2, 3
            â”‚   â”‚   â”œâ”€â•¼ 4 â•¾ 1, 2
            â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- add_edge(4, 2) ---
        â•™â”€â”€ 3 â•¾ 1, 2, 3
            â”œâ”€â•¼ 1 â•¾ 1, 2, 4
            â”‚   â”œâ”€â•¼ 2 â•¾ 2, 3, 4
            â”‚   â”‚   â”œâ”€â•¼ 4 â•¾ 1, 3
            â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- add_edge(4, 3) ---
        â•™â”€â”€ 4 â•¾ 1, 2, 3
            â”œâ”€â•¼ 1 â•¾ 1, 2, 3
            â”‚   â”œâ”€â•¼ 2 â•¾ 2, 3, 4
            â”‚   â”‚   â”œâ”€â•¼ 3 â•¾ 1, 3, 4
            â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- add_edge(4, 4) ---
        â•™â”€â”€ 1 â•¾ 1, 2, 3, 4
            â”œâ”€â•¼ 2 â•¾ 2, 3, 4
            â”‚   â”œâ”€â•¼ 3 â•¾ 1, 3, 4
            â”‚   â”‚   â”œâ”€â•¼ 4 â•¾ 1, 2, 4
            â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        N)r   r   r.   r   r   r   r   r/   r   r   r   ©r   r0   r1   ÚiÚjr2   r3   r   r   r   Ú4test_write_network_text_iterative_add_directed_edges5  s$   
ÿ rR   c                  C   s²   t  ¡ } |  g d¢¡ g }|j}|dƒ t j| |dd t| j| jƒD ]"\}}||kr,q#|d|› d|› dƒ |  ||¡ t j| |dd q#d |¡}t	d	ƒ 
¡ }||ksWJ ‚d
S )rJ   r'   rK   r   r+   rL   rM   rN   r   uø  
        --- initial state ---
        â•Ÿâ”€â”€ 1
        â•Ÿâ”€â”€ 2
        â•Ÿâ”€â”€ 3
        â•™â”€â”€ 4
        --- add_edge(1, 2) ---
        â•Ÿâ”€â”€ 3
        â•Ÿâ”€â”€ 4
        â•™â”€â”€ 1
            â””â”€â”€ 2
        --- add_edge(1, 3) ---
        â•Ÿâ”€â”€ 4
        â•™â”€â”€ 2
            â””â”€â”€ 1
                â””â”€â”€ 3
        --- add_edge(1, 4) ---
        â•™â”€â”€ 2
            â””â”€â”€ 1
                â”œâ”€â”€ 3
                â””â”€â”€ 4
        --- add_edge(2, 1) ---
        â•™â”€â”€ 2
            â””â”€â”€ 1
                â”œâ”€â”€ 3
                â””â”€â”€ 4
        --- add_edge(2, 3) ---
        â•™â”€â”€ 4
            â””â”€â”€ 1
                â”œâ”€â”€ 2
                â”‚   â””â”€â”€ 3 â”€ 1
                â””â”€â”€  ...
        --- add_edge(2, 4) ---
        â•™â”€â”€ 3
            â”œâ”€â”€ 1
            â”‚   â”œâ”€â”€ 2 â”€ 3
            â”‚   â”‚   â””â”€â”€ 4 â”€ 1
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        --- add_edge(3, 1) ---
        â•™â”€â”€ 3
            â”œâ”€â”€ 1
            â”‚   â”œâ”€â”€ 2 â”€ 3
            â”‚   â”‚   â””â”€â”€ 4 â”€ 1
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        --- add_edge(3, 2) ---
        â•™â”€â”€ 3
            â”œâ”€â”€ 1
            â”‚   â”œâ”€â”€ 2 â”€ 3
            â”‚   â”‚   â””â”€â”€ 4 â”€ 1
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        --- add_edge(3, 4) ---
        â•™â”€â”€ 1
            â”œâ”€â”€ 2
            â”‚   â”œâ”€â”€ 3 â”€ 1
            â”‚   â”‚   â””â”€â”€ 4 â”€ 1, 2
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        --- add_edge(4, 1) ---
        â•™â”€â”€ 1
            â”œâ”€â”€ 2
            â”‚   â”œâ”€â”€ 3 â”€ 1
            â”‚   â”‚   â””â”€â”€ 4 â”€ 1, 2
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        --- add_edge(4, 2) ---
        â•™â”€â”€ 1
            â”œâ”€â”€ 2
            â”‚   â”œâ”€â”€ 3 â”€ 1
            â”‚   â”‚   â””â”€â”€ 4 â”€ 1, 2
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        --- add_edge(4, 3) ---
        â•™â”€â”€ 1
            â”œâ”€â”€ 2
            â”‚   â”œâ”€â”€ 3 â”€ 1
            â”‚   â”‚   â””â”€â”€ 4 â”€ 1, 2
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        N)r   r%   r.   r   r   r   r   r/   r   r   r   rO   r   r   r   Ú6test_write_network_text_iterative_add_undirected_edgesÈ  s&   
ÿS­TrS   c            	      C   sÞ   t  d¡} t ¡ }| g d¢¡ tt|j|jƒƒ}|  |¡ | 	|dd… ¡ g }|j
}|dƒ tj||dd |dd… D ]\}}|d	|› d
|› dƒ | ||¡ tj||dd q>d |¡}tdƒ ¡ }||ksmJ ‚dS )rJ   i°y.+)r(   r   r)   r*   é   r   é   rK   r   r+   é   rL   rM   rN   r   u7  
        --- initial state ---
        â•™â”€â”€ 3 â•¾ 5
            â””â”€â•¼ 2 â•¾ 2
                â”œâ”€â•¼ 4 â•¾ 4
                â”‚   â”œâ”€â•¼ 5
                â”‚   â”‚   â”œâ”€â•¼ 1 â•¾ 1
                â”‚   â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â””â”€â•¼  ...
                â””â”€â•¼  ...
        --- add_edge(4, 1) ---
        â•™â”€â”€ 3 â•¾ 5
            â””â”€â•¼ 2 â•¾ 2
                â”œâ”€â•¼ 4 â•¾ 4
                â”‚   â”œâ”€â•¼ 5
                â”‚   â”‚   â”œâ”€â•¼ 1 â•¾ 1, 4
                â”‚   â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â””â”€â•¼  ...
                â””â”€â•¼  ...
        --- add_edge(2, 1) ---
        â•™â”€â”€ 3 â•¾ 5
            â””â”€â•¼ 2 â•¾ 2
                â”œâ”€â•¼ 4 â•¾ 4
                â”‚   â”œâ”€â•¼ 5
                â”‚   â”‚   â”œâ”€â•¼ 1 â•¾ 1, 4, 2
                â”‚   â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â””â”€â•¼  ...
                â””â”€â•¼  ...
        --- add_edge(5, 2) ---
        â•™â”€â”€ 3 â•¾ 5
            â””â”€â•¼ 2 â•¾ 2, 5
                â”œâ”€â•¼ 4 â•¾ 4
                â”‚   â”œâ”€â•¼ 5
                â”‚   â”‚   â”œâ”€â•¼ 1 â•¾ 1, 4, 2
                â”‚   â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â””â”€â•¼  ...
                â””â”€â•¼  ...
        --- add_edge(1, 5) ---
        â•™â”€â”€ 3 â•¾ 5
            â””â”€â•¼ 2 â•¾ 2, 5
                â”œâ”€â•¼ 4 â•¾ 4
                â”‚   â”œâ”€â•¼ 5 â•¾ 1
                â”‚   â”‚   â”œâ”€â•¼ 1 â•¾ 1, 4, 2
                â”‚   â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â””â”€â•¼  ...
                â””â”€â•¼  ...

        N)ÚrandomÚRandomr   r   r.   Úlistr   r   ÚshuffleZadd_edges_fromr   r   r/   r   r   r   )	Úrngr   Zpossible_edgesr0   r1   rP   rQ   r2   r3   r   r   r   Ú;test_write_network_text_iterative_add_random_directed_edges0  s*   


ÿ6Ê7r\   c                  C   s(  t  ¡ } |  dd¡ |  dd¡ |  dd¡ |  dd¡ |  dd¡ |  dd¡ |  dd¡ |  ¡ }|  dd¡ |  dd¡ |  dd¡ g }|j}|d	ƒ t j||d
d |dƒ t j| |d
d |dƒ t j| ¡ |dgd
d |dƒ t j|  ¡ |dgd
d d |¡}tdƒ 	¡ }||ks’J ‚d S )Nr(   r   rT   r)   r*   é   r6   rU   ú--- directed case ---r   r+   z"--- add (1, 8), (4, 2), (6, 3) ---ú--- undirected case ---©r,   r8   r   r   u  
        --- directed case ---
        â•™â”€â”€ 1
            â”œâ”€â•¼ 2
            â”‚   â””â”€â•¼ 3
            â”‚       â””â”€â•¼ 4
            â””â”€â•¼ 5
                â””â”€â•¼ 6
                    â”œâ”€â•¼ 7
                    â””â”€â•¼ 8
        --- add (1, 8), (4, 2), (6, 3) ---
        â•™â”€â”€ 1
            â”œâ”€â•¼ 2 â•¾ 4
            â”‚   â””â”€â•¼ 3 â•¾ 6
            â”‚       â””â”€â•¼ 4
            â”‚           â””â”€â•¼  ...
            â”œâ”€â•¼ 5
            â”‚   â””â”€â•¼ 6
            â”‚       â”œâ”€â•¼ 7
            â”‚       â”œâ”€â•¼ 8 â•¾ 1
            â”‚       â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- undirected case ---
        â•™â”€â”€ 1
            â”œâ”€â”€ 2
            â”‚   â””â”€â”€ 3
            â”‚       â””â”€â”€ 4
            â””â”€â”€ 5
                â””â”€â”€ 6
                    â”œâ”€â”€ 7
                    â””â”€â”€ 8
        --- add (1, 8), (4, 2), (6, 3) ---
        â•™â”€â”€ 1
            â”œâ”€â”€ 2
            â”‚   â”œâ”€â”€ 3
            â”‚   â”‚   â”œâ”€â”€ 4 â”€ 2
            â”‚   â”‚   â””â”€â”€ 6
            â”‚   â”‚       â”œâ”€â”€ 5 â”€ 1
            â”‚   â”‚       â”œâ”€â”€ 7
            â”‚   â”‚       â””â”€â”€ 8 â”€ 1
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        )
r   r   r/   Úcopyr   r   Úto_undirectedr   r   r   )r    Úorigr0   r1   r2   r3   r   r   r   Ú%test_write_network_text_nearly_forest~  s:   
ÿ,Ô-rd   c                  C   sz   t jjdt jd} g }|j}|dƒ t j| |ddd |dƒ t j|  ¡ |ddd d |¡}td	ƒ 	¡ }||ks;J ‚d S )
NrT   ©r   r^   Tr   r-   r_   r   aN  
        --- directed case ---
        +-- 0 <- 1, 2, 3, 4
            |-> 1 <- 2, 3, 4
            |   |-> 2 <- 0, 3, 4
            |   |   |-> 3 <- 0, 1, 4
            |   |   |   |-> 4 <- 0, 1, 2
            |   |   |   |   L->  ...
            |   |   |   L->  ...
            |   |   L->  ...
            |   L->  ...
            L->  ...
        --- undirected case ---
        +-- 0
            |-- 1
            |   |-- 2 - 0
            |   |   |-- 3 - 0, 1
            |   |   |   L-- 4 - 0, 1, 2
            |   |   L--  ...
            |   L--  ...
            L--  ...
        )
r   Ú
generatorsÚcomplete_graphr   r   r   rb   r   r   r   ©r   r0   r1   r2   r3   r   r   r   Ú1test_write_network_text_complete_graph_ascii_onlyÆ  s   
ÿéri   c                  C   sx   t jjdt jd} | jD ]}d|› d| j| d< qg }|j}t j| |dddd	 d
 |¡}tdƒ 	¡ }||ks:J ‚d S )NrT   re   zNode(n=ú)r
   TFr   )r,   r   r#   r   r   uK  
        â•™â”€â”€ Node(n=0) â•¾ Node(n=1), Node(n=2), Node(n=3), Node(n=4)
            â”œâ”€â•¼ Node(n=1) â•¾ Node(n=2), Node(n=3), Node(n=4)
            â”‚   â”œâ”€â•¼ Node(n=2) â•¾ Node(n=0), Node(n=3), Node(n=4)
            â”‚   â”‚   â”œâ”€â•¼ Node(n=3) â•¾ Node(n=0), Node(n=1), Node(n=4)
            â”‚   â”‚   â”‚   â”œâ”€â•¼ Node(n=4) â•¾ Node(n=0), Node(n=1), Node(n=2)
            â”‚   â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        )
r   rf   rg   r   r   r   r   r   r   r   )r   rB   r0   r1   r2   r3   r   r   r   Ú#test_write_network_text_with_labelsê  s   

ÿórk   c                  C   s    g } | j }dD ]}tj |¡}|d|› dƒ tj||dd qdD ]}tj |tj¡}|d|› dƒ tj||dd q"d | ¡}tdƒ ¡ }||ksNJ ‚d S )	N)r   r(   r   r)   r*   rT   z--- undirected k=ú ---r   r+   z--- directed k=r   ué  
        --- undirected k=0 ---
        â•™
        --- undirected k=1 ---
        â•™â”€â”€ 0
        --- undirected k=2 ---
        â•™â”€â”€ 0
            â””â”€â”€ 1
        --- undirected k=3 ---
        â•™â”€â”€ 0
            â”œâ”€â”€ 1
            â”‚   â””â”€â”€ 2 â”€ 0
            â””â”€â”€  ...
        --- undirected k=4 ---
        â•™â”€â”€ 0
            â”œâ”€â”€ 1
            â”‚   â”œâ”€â”€ 2 â”€ 0
            â”‚   â”‚   â””â”€â”€ 3 â”€ 0, 1
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        --- undirected k=5 ---
        â•™â”€â”€ 0
            â”œâ”€â”€ 1
            â”‚   â”œâ”€â”€ 2 â”€ 0
            â”‚   â”‚   â”œâ”€â”€ 3 â”€ 0, 1
            â”‚   â”‚   â”‚   â””â”€â”€ 4 â”€ 0, 1, 2
            â”‚   â”‚   â””â”€â”€  ...
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        --- directed k=0 ---
        â•™
        --- directed k=1 ---
        â•™â”€â”€ 0
        --- directed k=2 ---
        â•™â”€â”€ 0 â•¾ 1
            â””â”€â•¼ 1
                â””â”€â•¼  ...
        --- directed k=3 ---
        â•™â”€â”€ 0 â•¾ 1, 2
            â”œâ”€â•¼ 1 â•¾ 2
            â”‚   â”œâ”€â•¼ 2 â•¾ 0
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- directed k=4 ---
        â•™â”€â”€ 0 â•¾ 1, 2, 3
            â”œâ”€â•¼ 1 â•¾ 2, 3
            â”‚   â”œâ”€â•¼ 2 â•¾ 0, 3
            â”‚   â”‚   â”œâ”€â•¼ 3 â•¾ 0, 1
            â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- directed k=5 ---
        â•™â”€â”€ 0 â•¾ 1, 2, 3, 4
            â”œâ”€â•¼ 1 â•¾ 2, 3, 4
            â”‚   â”œâ”€â•¼ 2 â•¾ 0, 3, 4
            â”‚   â”‚   â”œâ”€â•¼ 3 â•¾ 0, 1, 4
            â”‚   â”‚   â”‚   â”œâ”€â•¼ 4 â•¾ 0, 1, 2
            â”‚   â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        )	r   r   rf   rg   r   r   r   r   r   )r0   r1   Úkr    r2   r3   r   r   r   Ú'test_write_network_text_complete_graphs  s"   
ÿB¾Crn   c                  C   sÒ   t  ¡ } |  dd¡ |  dd¡ |  dd¡ |  dd¡ |  dd¡ |  dd¡ |  dd¡ |  dd¡ g }|j}t|  ¡ ƒ}|D ]}|d|› dƒ t j| ||gd	d
 qAd |¡}tdƒ 	¡ }||ksgJ ‚d S )Nr(   r   r)   r*   rT   r]   z--- source node: rl   r   r`   r   ux  
        --- source node: 1 ---
        â•™â”€â”€ 1 â•¾ 4
            â”œâ”€â•¼ 2
            â”‚   â””â”€â•¼ 4 â•¾ 5
            â”‚       â””â”€â•¼  ...
            â”œâ”€â•¼ 3
            â”‚   â”œâ”€â•¼ 5 â•¾ 1
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼ 6
            â””â”€â•¼  ...
        --- source node: 2 ---
        â•™â”€â”€ 2 â•¾ 1
            â””â”€â•¼ 4 â•¾ 5
                â””â”€â•¼ 1
                    â”œâ”€â•¼ 3
                    â”‚   â”œâ”€â•¼ 5 â•¾ 1
                    â”‚   â”‚   â””â”€â•¼  ...
                    â”‚   â””â”€â•¼ 6
                    â””â”€â•¼  ...
        --- source node: 3 ---
        â•™â”€â”€ 3 â•¾ 1
            â”œâ”€â•¼ 5 â•¾ 1
            â”‚   â””â”€â•¼ 4 â•¾ 2
            â”‚       â””â”€â•¼ 1
            â”‚           â”œâ”€â•¼ 2
            â”‚           â”‚   â””â”€â•¼  ...
            â”‚           â””â”€â•¼  ...
            â””â”€â•¼ 6
        --- source node: 4 ---
        â•™â”€â”€ 4 â•¾ 2, 5
            â””â”€â•¼ 1
                â”œâ”€â•¼ 2
                â”‚   â””â”€â•¼  ...
                â”œâ”€â•¼ 3
                â”‚   â”œâ”€â•¼ 5 â•¾ 1
                â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â””â”€â•¼ 6
                â””â”€â•¼  ...
        --- source node: 5 ---
        â•™â”€â”€ 5 â•¾ 3, 1
            â””â”€â•¼ 4 â•¾ 2
                â””â”€â•¼ 1
                    â”œâ”€â•¼ 2
                    â”‚   â””â”€â•¼  ...
                    â”œâ”€â•¼ 3
                    â”‚   â”œâ”€â•¼ 6
                    â”‚   â””â”€â•¼  ...
                    â””â”€â•¼  ...
        --- source node: 6 ---
        â•™â”€â”€ 6 â•¾ 3
        )
r   r   r/   r   Úsortedr   r   r   r   r   )r    r0   r1   r   rB   r2   r3   r   r   r   Ú(test_write_network_text_multiple_sourcesX  s,   
ÿ5Ë6rp   c                  C   óP   t jdt jd} g }|j}t j| |dd d |¡}tdƒ ¡ }||ks&J ‚d S )NrT   re   r   r+   r   u¥   
        â•™â”€â”€ 1
            â””â”€â”€ 0
                â”œâ”€â”€ 2
                â”œâ”€â”€ 3
                â”œâ”€â”€ 4
                â””â”€â”€ 5
        )r   Ú
star_graphr%   r   r   r   r   r   rh   r   r   r   Ú"test_write_network_text_star_graph¤  s   
ÿ	÷
rs   c                  C   rq   )Nr)   re   r   r+   r   uQ   
        â•™â”€â”€ 0
            â””â”€â”€ 1
                â””â”€â”€ 2
        )r   Ú
path_graphr%   r   r   r   r   r   rh   r   r   r   Ú"test_write_network_text_path_graph·  s   
ÿúru   c                  C   sR   t jddt jd} g }|j}t j| |dd d |¡}tdƒ ¡ }||ks'J ‚d S )Nr*   r   re   r   r+   r   u  
        â•™â”€â”€ 5
            â””â”€â”€ 4
                â””â”€â”€ 3
                    â”œâ”€â”€ 0
                    â”‚   â”œâ”€â”€ 1 â”€ 3
                    â”‚   â”‚   â””â”€â”€ 2 â”€ 0, 3
                    â”‚   â””â”€â”€  ...
                    â””â”€â”€  ...
        )r   Úlollipop_graphr%   r   r   r   r   r   rh   r   r   r   Ú&test_write_network_text_lollipop_graphÇ  s   
ÿõrw   c                  C   rq   )Nr6   re   r   r+   r   uX  
        â•™â”€â”€ 1
            â”œâ”€â”€ 0
            â”‚   â”œâ”€â”€ 2 â”€ 1
            â”‚   â”‚   â””â”€â”€ 3 â”€ 0
            â”‚   â”‚       â””â”€â”€ 4 â”€ 0
            â”‚   â”‚           â””â”€â”€ 5 â”€ 0
            â”‚   â”‚               â””â”€â”€ 6 â”€ 0, 1
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        )r   Úwheel_graphr%   r   r   r   r   r   rh   r   r   r   Ú#test_write_network_text_wheel_graphÜ  s   
ÿôry   c                  C   rq   )Nr*   re   r   r+   r   uê  
        â•™â”€â”€ 0
            â”œâ”€â”€ 1
            â”‚   â”œâ”€â”€ 2
            â”‚   â”‚   â”œâ”€â”€ 3 â”€ 0
            â”‚   â”‚   â”‚   â””â”€â”€ 7
            â”‚   â”‚   â”‚       â”œâ”€â”€ 6 â”€ 2
            â”‚   â”‚   â”‚       â”‚   â””â”€â”€ 5 â”€ 1
            â”‚   â”‚   â”‚       â”‚       â””â”€â”€ 4 â”€ 0, 7
            â”‚   â”‚   â”‚       â””â”€â”€  ...
            â”‚   â”‚   â””â”€â”€  ...
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        )r   Úcircular_ladder_graphr%   r   r   r   r   r   rh   r   r   r   Ú-test_write_network_text_circular_ladder_graphò  s   
ÿñr{   c                  C   rq   )Nr*   re   r   r+   r   u›	  
        â•™â”€â”€ 15
            â”œâ”€â”€ 0
            â”‚   â”œâ”€â”€ 1 â”€ 15
            â”‚   â”‚   â”œâ”€â”€ 2 â”€ 0
            â”‚   â”‚   â”‚   â”œâ”€â”€ 4 â”€ 0
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 9 â”€ 0
            â”‚   â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 22 â”€ 0
            â”‚   â”‚   â”‚   â”‚   â”‚   â””â”€â”€ 38 â”€ 4
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 13 â”€ 2
            â”‚   â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 34 â”€ 2
            â”‚   â”‚   â”‚   â”‚   â”‚   â””â”€â”€ 39 â”€ 4
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 18 â”€ 0
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 30 â”€ 2
            â”‚   â”‚   â”‚   â”‚   â””â”€â”€  ...
            â”‚   â”‚   â”‚   â”œâ”€â”€ 5 â”€ 1
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 12 â”€ 1
            â”‚   â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 29 â”€ 1
            â”‚   â”‚   â”‚   â”‚   â”‚   â””â”€â”€ 40 â”€ 5
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 14 â”€ 2
            â”‚   â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 35 â”€ 2
            â”‚   â”‚   â”‚   â”‚   â”‚   â””â”€â”€ 41 â”€ 5
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 25 â”€ 1
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 31 â”€ 2
            â”‚   â”‚   â”‚   â”‚   â””â”€â”€  ...
            â”‚   â”‚   â”‚   â”œâ”€â”€ 7 â”€ 0
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 20 â”€ 0
            â”‚   â”‚   â”‚   â”‚   â””â”€â”€ 32 â”€ 2
            â”‚   â”‚   â”‚   â”œâ”€â”€ 10 â”€ 1
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 27 â”€ 1
            â”‚   â”‚   â”‚   â”‚   â””â”€â”€ 33 â”€ 2
            â”‚   â”‚   â”‚   â”œâ”€â”€ 16 â”€ 0
            â”‚   â”‚   â”‚   â”œâ”€â”€ 23 â”€ 1
            â”‚   â”‚   â”‚   â””â”€â”€  ...
            â”‚   â”‚   â”œâ”€â”€ 3 â”€ 0
            â”‚   â”‚   â”‚   â”œâ”€â”€ 8 â”€ 0
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 21 â”€ 0
            â”‚   â”‚   â”‚   â”‚   â””â”€â”€ 36 â”€ 3
            â”‚   â”‚   â”‚   â”œâ”€â”€ 11 â”€ 1
            â”‚   â”‚   â”‚   â”‚   â”œâ”€â”€ 28 â”€ 1
            â”‚   â”‚   â”‚   â”‚   â””â”€â”€ 37 â”€ 3
            â”‚   â”‚   â”‚   â”œâ”€â”€ 17 â”€ 0
            â”‚   â”‚   â”‚   â”œâ”€â”€ 24 â”€ 1
            â”‚   â”‚   â”‚   â””â”€â”€  ...
            â”‚   â”‚   â”œâ”€â”€ 6 â”€ 0
            â”‚   â”‚   â”‚   â”œâ”€â”€ 19 â”€ 0
            â”‚   â”‚   â”‚   â””â”€â”€ 26 â”€ 1
            â”‚   â”‚   â””â”€â”€  ...
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        )r   Ú dorogovtsev_goltsev_mendes_graphr%   r   r   r   r   r   rh   r   r   r   Ú8test_write_network_text_dorogovtsev_goltsev_mendes_graph  s   
ÿ4Ì5r}   c                  C   sZ  t jddt jd} g }|j}|dƒ t j| |ddd |dƒ t j| |ddd |d	ƒ t j| |dd
d |dƒ t j| |ddd |dƒ t j| |ddd |dƒ t j|  ¡ |ddd |dƒ t j|  ¡ |ddd |dƒ t j|  ¡ |dd
d |dƒ t j|  ¡ |ddd |dƒ t j|  ¡ |ddd d |¡}tdƒ ¡ }||ks«J ‚d S )Nr(   r)   r   ú"--- directed case, max_depth=0 ---r   r   ©r,   r   Ú	max_depthú"--- directed case, max_depth=1 ---ú"--- directed case, max_depth=2 ---r   ú"--- directed case, max_depth=3 ---z"--- directed case, max_depth=4 ---r*   ú$--- undirected case, max_depth=0 ---ú$--- undirected case, max_depth=1 ---ú$--- undirected case, max_depth=2 ---ú$--- undirected case, max_depth=3 ---z$--- undirected case, max_depth=4 ---r   u  
        --- directed case, max_depth=0 ---
        â•™ ...
        --- directed case, max_depth=1 ---
        â•™â”€â”€ 0
            â””â”€â•¼  ...
        --- directed case, max_depth=2 ---
        â•™â”€â”€ 0
            â””â”€â•¼ 1
                â””â”€â•¼  ...
        --- directed case, max_depth=3 ---
        â•™â”€â”€ 0
            â””â”€â•¼ 1
                â””â”€â•¼ 2
                    â””â”€â•¼  ...
        --- directed case, max_depth=4 ---
        â•™â”€â”€ 0
            â””â”€â•¼ 1
                â””â”€â•¼ 2
                    â””â”€â•¼ 3
        --- undirected case, max_depth=0 ---
        â•™ ...
        --- undirected case, max_depth=1 ---
        â•™â”€â”€ 0 â”€ 1
            â””â”€â”€  ...
        --- undirected case, max_depth=2 ---
        â•™â”€â”€ 0
            â””â”€â”€ 1 â”€ 2
                â””â”€â”€  ...
        --- undirected case, max_depth=3 ---
        â•™â”€â”€ 0
            â””â”€â”€ 1
                â””â”€â”€ 2 â”€ 3
                    â””â”€â”€  ...
        --- undirected case, max_depth=4 ---
        â•™â”€â”€ 0
            â””â”€â”€ 1
                â””â”€â”€ 2
                    â””â”€â”€ 3
        )	r   r   r   r   r   rb   r   r   r   ©rc   r0   r1   r2   r3   r   r   r   Ú&test_write_network_text_tree_max_depthI  s<   
ÿ)×*r‰   c                  C   sZ  t jddddd} g }|j}|dƒ t j| |dd d |d	ƒ t j| |dd
d |dƒ t j| |ddd |dƒ t j| |ddd |dƒ t j| |ddd |dƒ t j|  ¡ |dd d |dƒ t j|  ¡ |dd
d |dƒ t j|  ¡ |ddd |dƒ t j|  ¡ |ddd |dƒ t j|  ¡ |ddd d |¡}tdƒ ¡ }||ks«J ‚d S )Né
   g333333Ã?TiÈ  ©ÚdirectedÚseedú%--- directed case, max_depth=None ---r   r   r~   r   r   r(   r‚   r   rƒ   r)   ú'--- undirected case, max_depth=None ---r„   r…   r†   r‡   r   uê	  
        --- directed case, max_depth=None ---
        â•Ÿâ”€â”€ 4
        â•Ž   â”œâ”€â•¼ 0 â•¾ 3
        â•Ž   â”œâ”€â•¼ 5 â•¾ 7
        â•Ž   â”‚   â””â”€â•¼ 3
        â•Ž   â”‚       â”œâ”€â•¼ 1 â•¾ 9
        â•Ž   â”‚       â”‚   â””â”€â•¼ 9 â•¾ 6
        â•Ž   â”‚       â”‚       â”œâ”€â•¼ 6
        â•Ž   â”‚       â”‚       â”‚   â””â”€â•¼  ...
        â•Ž   â”‚       â”‚       â”œâ”€â•¼ 7 â•¾ 4
        â•Ž   â”‚       â”‚       â”‚   â”œâ”€â•¼ 2
        â•Ž   â”‚       â”‚       â”‚   â””â”€â•¼  ...
        â•Ž   â”‚       â”‚       â””â”€â•¼  ...
        â•Ž   â”‚       â””â”€â•¼  ...
        â•Ž   â””â”€â•¼  ...
        â•™â”€â”€ 8
        --- directed case, max_depth=0 ---
        â•™ ...
        --- directed case, max_depth=1 ---
        â•Ÿâ”€â”€ 4
        â•Ž   â””â”€â•¼  ...
        â•™â”€â”€ 8
        --- directed case, max_depth=2 ---
        â•Ÿâ”€â”€ 4
        â•Ž   â”œâ”€â•¼ 0 â•¾ 3
        â•Ž   â”œâ”€â•¼ 5 â•¾ 7
        â•Ž   â”‚   â””â”€â•¼  ...
        â•Ž   â””â”€â•¼ 7 â•¾ 9
        â•Ž       â””â”€â•¼  ...
        â•™â”€â”€ 8
        --- directed case, max_depth=3 ---
        â•Ÿâ”€â”€ 4
        â•Ž   â”œâ”€â•¼ 0 â•¾ 3
        â•Ž   â”œâ”€â•¼ 5 â•¾ 7
        â•Ž   â”‚   â””â”€â•¼ 3
        â•Ž   â”‚       â””â”€â•¼  ...
        â•Ž   â””â”€â•¼ 7 â•¾ 9
        â•Ž       â”œâ”€â•¼ 2
        â•Ž       â””â”€â•¼  ...
        â•™â”€â”€ 8
        --- undirected case, max_depth=None ---
        â•Ÿâ”€â”€ 8
        â•™â”€â”€ 2
            â””â”€â”€ 7
                â”œâ”€â”€ 4
                â”‚   â”œâ”€â”€ 0
                â”‚   â”‚   â””â”€â”€ 3
                â”‚   â”‚       â”œâ”€â”€ 1
                â”‚   â”‚       â”‚   â””â”€â”€ 9 â”€ 7
                â”‚   â”‚       â”‚       â””â”€â”€ 6
                â”‚   â”‚       â””â”€â”€ 5 â”€ 4, 7
                â”‚   â””â”€â”€  ...
                â””â”€â”€  ...
        --- undirected case, max_depth=0 ---
        â•™ ...
        --- undirected case, max_depth=1 ---
        â•Ÿâ”€â”€ 8
        â•™â”€â”€ 2 â”€ 7
            â””â”€â”€  ...
        --- undirected case, max_depth=2 ---
        â•Ÿâ”€â”€ 8
        â•™â”€â”€ 2
            â””â”€â”€ 7 â”€ 4, 5, 9
                â””â”€â”€  ...
        --- undirected case, max_depth=3 ---
        â•Ÿâ”€â”€ 8
        â•™â”€â”€ 2
            â””â”€â”€ 7
                â”œâ”€â”€ 4 â”€ 0, 5
                â”‚   â””â”€â”€  ...
                â”œâ”€â”€ 5 â”€ 4, 3
                â”‚   â””â”€â”€  ...
                â””â”€â”€ 9 â”€ 1, 6
                    â””â”€â”€  ...
        )r   Úerdos_renyi_graphr   r   rb   r   r   r   rˆ   r   r   r   Ú'test_write_network_text_graph_max_depth  s<   
ÿM³Nr‘   c                  C   sV  t  dt j¡} g }|j}|dƒ t j| |dd d |dƒ t j| |ddd |dƒ t j| |ddd |d	ƒ t j| |dd
d |dƒ t j| |ddd |dƒ t j|  ¡ |dd d |dƒ t j|  ¡ |ddd |dƒ t j|  ¡ |ddd |dƒ t j|  ¡ |dd
d |dƒ t j|  ¡ |ddd d |¡}tdƒ ¡ }||ks©J ‚d S )NrT   rŽ   r   r   r~   r   r   r(   r‚   r   rƒ   r)   r   r„   r…   r†   r‡   r   uØ	  
        --- directed case, max_depth=None ---
        â•™â”€â”€ 0 â•¾ 1, 2, 3, 4
            â”œâ”€â•¼ 1 â•¾ 2, 3, 4
            â”‚   â”œâ”€â•¼ 2 â•¾ 0, 3, 4
            â”‚   â”‚   â”œâ”€â•¼ 3 â•¾ 0, 1, 4
            â”‚   â”‚   â”‚   â”œâ”€â•¼ 4 â•¾ 0, 1, 2
            â”‚   â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- directed case, max_depth=0 ---
        â•™ ...
        --- directed case, max_depth=1 ---
        â•™â”€â”€ 0 â•¾ 1, 2, 3, 4
            â””â”€â•¼  ...
        --- directed case, max_depth=2 ---
        â•™â”€â”€ 0 â•¾ 1, 2, 3, 4
            â”œâ”€â•¼ 1 â•¾ 2, 3, 4
            â”‚   â””â”€â•¼  ...
            â”œâ”€â•¼ 2 â•¾ 1, 3, 4
            â”‚   â””â”€â•¼  ...
            â”œâ”€â•¼ 3 â•¾ 1, 2, 4
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼ 4 â•¾ 1, 2, 3
                â””â”€â•¼  ...
        --- directed case, max_depth=3 ---
        â•™â”€â”€ 0 â•¾ 1, 2, 3, 4
            â”œâ”€â•¼ 1 â•¾ 2, 3, 4
            â”‚   â”œâ”€â•¼ 2 â•¾ 0, 3, 4
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”œâ”€â•¼ 3 â•¾ 0, 2, 4
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â”œâ”€â•¼ 4 â•¾ 0, 2, 3
            â”‚   â”‚   â””â”€â•¼  ...
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼  ...
        --- undirected case, max_depth=None ---
        â•™â”€â”€ 0
            â”œâ”€â”€ 1
            â”‚   â”œâ”€â”€ 2 â”€ 0
            â”‚   â”‚   â”œâ”€â”€ 3 â”€ 0, 1
            â”‚   â”‚   â”‚   â””â”€â”€ 4 â”€ 0, 1, 2
            â”‚   â”‚   â””â”€â”€  ...
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        --- undirected case, max_depth=0 ---
        â•™ ...
        --- undirected case, max_depth=1 ---
        â•™â”€â”€ 0 â”€ 1, 2, 3, 4
            â””â”€â”€  ...
        --- undirected case, max_depth=2 ---
        â•™â”€â”€ 0
            â”œâ”€â”€ 1 â”€ 2, 3, 4
            â”‚   â””â”€â”€  ...
            â”œâ”€â”€ 2 â”€ 1, 3, 4
            â”‚   â””â”€â”€  ...
            â”œâ”€â”€ 3 â”€ 1, 2, 4
            â”‚   â””â”€â”€  ...
            â””â”€â”€ 4 â”€ 1, 2, 3
        --- undirected case, max_depth=3 ---
        â•™â”€â”€ 0
            â”œâ”€â”€ 1
            â”‚   â”œâ”€â”€ 2 â”€ 0, 3, 4
            â”‚   â”‚   â””â”€â”€  ...
            â”‚   â”œâ”€â”€ 3 â”€ 0, 2, 4
            â”‚   â”‚   â””â”€â”€  ...
            â”‚   â””â”€â”€ 4 â”€ 0, 2, 3
            â””â”€â”€  ...
        )	r   rg   r   r   r   rb   r   r   r   rˆ   r   r   r   Ú(test_write_network_text_clique_max_depthù  s<   
ÿH¸Ir’   c                  C   s  t jddddd} | jD ]/}d|› d| j| d< t|td	ƒ d
 ƒ| j| d< |d dkr;t|td	ƒ ƒ| j| d< qg }|j}|dƒ t j| |ddd d |dƒ t j| |ddd d |dƒ t j| |ddd d |dƒ t j| |ddd d d |¡}tdƒ 	¡ }||ksŠJ ‚d S )NrT   gš™™™™™Ù?TiVLir‹   zNode(rj   r
   r	   r(   r   r   r   Úpartz4--- when with_labels=True, uses the 'label' attr ---r   )r,   r   r   r€   z4--- when with_labels=False, uses str(node) value ---Fz3--- when with_labels is a string, use that attr ---z:--- fallback to str(node) when the attr does not exist ---r   uj  
        --- when with_labels=True, uses the 'label' attr ---
        â•™â”€â”€ Node(1)
            â””â”€â•¼ Node(3) â•¾ Node(2)
                â”œâ”€â•¼ Node(0)
                â”‚   â”œâ”€â•¼ Node(2) â•¾ Node(3), Node(4)
                â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â””â”€â•¼ Node(4)
                â”‚       â””â”€â•¼  ...
                â””â”€â•¼  ...
        --- when with_labels=False, uses str(node) value ---
        â•™â”€â”€ 1
            â””â”€â•¼ 3 â•¾ 2
                â”œâ”€â•¼ 0
                â”‚   â”œâ”€â•¼ 2 â•¾ 3, 4
                â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â””â”€â•¼ 4
                â”‚       â””â”€â•¼  ...
                â””â”€â•¼  ...
        --- when with_labels is a string, use that attr ---
        â•™â”€â”€ a
            â””â”€â•¼ c â•¾ b
                â”œâ”€â•¼ `
                â”‚   â”œâ”€â•¼ b â•¾ c, d
                â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â””â”€â•¼ d
                â”‚       â””â”€â•¼  ...
                â””â”€â•¼  ...
        --- fallback to str(node) when the attr does not exist ---
        â•™â”€â”€ 1
            â””â”€â•¼ 3 â•¾ c
                â”œâ”€â•¼ a
                â”‚   â”œâ”€â•¼ c â•¾ 3, e
                â”‚   â”‚   â””â”€â•¼  ...
                â”‚   â””â”€â•¼ e
                â”‚       â””â”€â•¼  ...
                â””â”€â•¼  ...
        )
r   r   r   r   r   r   r   r   r   r   )r   r   r0   r1   r2   r3   r   r   r   Ú$test_write_network_text_custom_label^  s0   
€
ÿ'Ù(r”   c                  C   sð   t jddt jd} |  dd¡ |  dd¡ |  dd¡ |  ¡ }| dd	„ |jD ƒ¡ g }|j}|d
ƒ t j| |ddd |dƒ t j| |dddd |dƒ t j||ddd |dƒ t j||dddd d 	|¡}t
dƒ ¡ }||ksvJ ‚d S )Nr*   r   re   r   éÿÿÿÿéþÿÿÿéýÿÿÿc                 S   s    g | ]\}}||kr||f‘qS r   r   )rA   ÚuÚvr   r   r   Ú
<listcomp>¥  s     z;test_write_network_text_vertical_chains.<locals>.<listcomp>z--- Undirected UTF ---r   T)r,   r   Úvertical_chainsz--- Undirected ASCI ---)r,   r   r›   r#   z--- Directed UTF ---z--- Directed ASCI ---r   uí  
        --- Undirected UTF ---
        â•™â”€â”€ 5
            â”‚
            4
            â”‚
            3
            â”œâ”€â”€ 0
            â”‚   â”œâ”€â”€ 1 â”€ 3
            â”‚   â”‚   â”‚
            â”‚   â”‚   2 â”€ 0, 3
            â”‚   â”œâ”€â”€ -1
            â”‚   â”‚   â”‚
            â”‚   â”‚   -2
            â”‚   â”‚   â”‚
            â”‚   â”‚   -3
            â”‚   â””â”€â”€  ...
            â””â”€â”€  ...
        --- Undirected ASCI ---
        +-- 5
            |
            4
            |
            3
            |-- 0
            |   |-- 1 - 3
            |   |   |
            |   |   2 - 0, 3
            |   |-- -1
            |   |   |
            |   |   -2
            |   |   |
            |   |   -3
            |   L--  ...
            L--  ...
        --- Directed UTF ---
        â•™â”€â”€ 5
            â•½
            4
            â•½
            3
            â”œâ”€â•¼ 0 â•¾ 1, 2
            â”‚   â•½
            â”‚   -1
            â”‚   â•½
            â”‚   -2
            â”‚   â•½
            â”‚   -3
            â”œâ”€â•¼ 1 â•¾ 2
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼ 2
                â””â”€â•¼  ...
        --- Directed ASCI ---
        +-- 5
            !
            4
            !
            3
            |-> 0 <- 1, 2
            |   !
            |   -1
            |   !
            |   -2
            |   !
            |   -3
            |-> 1 <- 2
            |   L->  ...
            L-> 2
                L->  ...
        )r   rv   r%   r/   Zto_directedZremove_edges_fromÚedgesr   r   r   r   r   )Zgraph1Zgraph2r0   r1   r2   r3   r   r   r   Ú'test_write_network_text_vertical_chainsž  s6   
ÿ
ÿ
ÿG¹Hr   c                  C   sÊ   t jddt jd} g }|j}|dƒ t j| |dd d| jd d	< |d
ƒ t j| |dd |dƒ |  dd¡ t j| |dd |dƒ d| jd d	< t j| |dd d |¡}tdƒ 	¡ }||kscJ ‚d S )Nr   r)   r   ú--- Original ---r   r+   Tr(   Úcollapseú--- Collapse Node 1 ---ú/--- Add alternate path (5, 3) to collapsed zonerT   ú--- Collapse Node 0 ---r   r   uT  
        --- Original ---
        â•™â”€â”€ 0
            â”œâ”€â•¼ 1
            â”‚   â”œâ”€â•¼ 3
            â”‚   â”‚   â”œâ”€â•¼ 7
            â”‚   â”‚   â””â”€â•¼ 8
            â”‚   â””â”€â•¼ 4
            â”‚       â”œâ”€â•¼ 9
            â”‚       â””â”€â•¼ 10
            â””â”€â•¼ 2
                â”œâ”€â•¼ 5
                â”‚   â”œâ”€â•¼ 11
                â”‚   â””â”€â•¼ 12
                â””â”€â•¼ 6
                    â”œâ”€â•¼ 13
                    â””â”€â•¼ 14
        --- Collapse Node 1 ---
        â•™â”€â”€ 0
            â”œâ”€â•¼ 1
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼ 2
                â”œâ”€â•¼ 5
                â”‚   â”œâ”€â•¼ 11
                â”‚   â””â”€â•¼ 12
                â””â”€â•¼ 6
                    â”œâ”€â•¼ 13
                    â””â”€â•¼ 14
        --- Add alternate path (5, 3) to collapsed zone
        â•™â”€â”€ 0
            â”œâ”€â•¼ 1
            â”‚   â””â”€â•¼  ...
            â””â”€â•¼ 2
                â”œâ”€â•¼ 5
                â”‚   â”œâ”€â•¼ 11
                â”‚   â”œâ”€â•¼ 12
                â”‚   â””â”€â•¼ 3 â•¾ 1
                â”‚       â”œâ”€â•¼ 7
                â”‚       â””â”€â•¼ 8
                â””â”€â•¼ 6
                    â”œâ”€â•¼ 13
                    â””â”€â•¼ 14
        --- Collapse Node 0 ---
        â•™â”€â”€ 0
            â””â”€â•¼  ...
        )
r   r   r   r   r   r   r/   r   r   r   rh   r   r   r   Útest_collapse_directed  s*   
ÿ/Ñ0r£   c                  C   sÚ   t jddt jd} g }|j}|dƒ t j| |ddgd d| jd	 d
< |dƒ t j| |ddgd |dƒ |  dd¡ t j| |ddgd |dƒ d| jd d
< t j| |ddgd d |¡}tdƒ 	¡ }||kskJ ‚d S )Nr   r)   r   rž   r   r   )r,   r   r8   Tr(   rŸ   r    r¡   rT   r¢   r   uo  
        --- Original ---
        â•™â”€â”€ 0
            â”œâ”€â”€ 1
            â”‚   â”œâ”€â”€ 3
            â”‚   â”‚   â”œâ”€â”€ 7
            â”‚   â”‚   â””â”€â”€ 8
            â”‚   â””â”€â”€ 4
            â”‚       â”œâ”€â”€ 9
            â”‚       â””â”€â”€ 10
            â””â”€â”€ 2
                â”œâ”€â”€ 5
                â”‚   â”œâ”€â”€ 11
                â”‚   â””â”€â”€ 12
                â””â”€â”€ 6
                    â”œâ”€â”€ 13
                    â””â”€â”€ 14
        --- Collapse Node 1 ---
        â•™â”€â”€ 0
            â”œâ”€â”€ 1 â”€ 3, 4
            â”‚   â””â”€â”€  ...
            â””â”€â”€ 2
                â”œâ”€â”€ 5
                â”‚   â”œâ”€â”€ 11
                â”‚   â””â”€â”€ 12
                â””â”€â”€ 6
                    â”œâ”€â”€ 13
                    â””â”€â”€ 14
        --- Add alternate path (5, 3) to collapsed zone
        â•™â”€â”€ 0
            â”œâ”€â”€ 1 â”€ 3, 4
            â”‚   â””â”€â”€  ...
            â””â”€â”€ 2
                â”œâ”€â”€ 5
                â”‚   â”œâ”€â”€ 11
                â”‚   â”œâ”€â”€ 12
                â”‚   â””â”€â”€ 3 â”€ 1
                â”‚       â”œâ”€â”€ 7
                â”‚       â””â”€â”€ 8
                â””â”€â”€ 6
                    â”œâ”€â”€ 13
                    â””â”€â”€ 14
        --- Collapse Node 0 ---
        â•™â”€â”€ 0 â”€ 1, 2
            â””â”€â”€  ...
        )
r   r   r%   r   r   r   r/   r   r   r   rh   r   r   r   Útest_collapse_undirectedE  s*   
ÿ/Ñ0r¤   c            	   
   c   s   ddl } |  d¡}d}dD ]}}|rtjntj}tdƒD ]2}|ƒ }| t|ƒ¡ |V  |dkrNdD ]}t|ƒD ]}tj||||d}|V  q7q1t ||¡V  qtj	d|d	V  tj
d
d|dV  |s‹tjd|d	V  tjd|d	V  tjdd|d	V  tjd|d	V  tjd|d	V  qdS )zP
    Generate a gauntlet of different test graphs with different properties
    r   Nip7:r)   ©r   r(   é   )gš™™™™™¹?g333333Ó?g      à?gffffffæ?gÍÌÌÌÌÌì?r‹   re   r(   r   r*   rT   r   r6   )rW   rX   r   r   r%   Úranger.   r   rg   rt   r   rz   rr   rv   rx   r|   )	rW   r[   Znum_randomizedrŒ   ÚclsZ	num_nodesr   Úpr   r   r   r   Úgenerate_test_graphsˆ  s<   €
ÿü€€årª   ©r›   r#   c                 C   s   g | ]}d D ]}||f‘qqS )r¥   r   )rA   r›   r#   r   r   r   rš   ²  s    ýþÿrš   r¥   c              	   C   sŽ   ddl m} tƒ D ];}t |dd„ |jD ƒ¡}ttj|| |dƒ}||ƒ}z|j|jks.J ‚|j|jks6J ‚W q	 t	yD   t 
|¡ ‚ w dS )zÍ
    Write the graph to network text format, then parse it back in, assert it is
    the same as the original graph. Passing this test is strong validation of
    both the format generator and parser.
    r   )Ú_parse_network_textc                 S   s   i | ]}|t |ƒ“qS r   )Ústrr@   r   r   r   rD   Â  s    z0test_network_text_round_trip.<locals>.<dictcomp>r«   N)Znetworkx.readwrite.textr¬   rª   r   rE   r   rY   r   rœ   Ú	Exceptionr   )r›   r#   r¬   r   r0   Únewr   r   r   Útest_network_text_round_trip¯  s"   
ÿÿ
þõr°   ))rW   Ú	itertoolsr   Útextwrapr   ZpytestZnetworkxr   r   r&   r4   r>   rG   rH   rI   rR   rS   r\   rd   ri   rk   rn   rp   rs   ru   rw   ry   r{   r}   r‰   r‘   r’   r”   r   r£   r¤   rª   ÚmarkZparametrizeÚtupler°   r   r   r   r   Ú<module>   sX    'Y2$1 hNH$SL>Fje@dCC'þÿþ
