
    ܛ7i'                     D    d dl Z d dlZd dlmZmZ d dlmZ  G d d      Zy)    N)read_p2g	write_p2g)edges_equalc                   .    e Zd Zed        Zd Zd Zd Zy)TestP2Gc                     t        j                  d      | _        g d}| j                  j                  |       | j                  j	                  d       t        j
                  | j                        | _        y )Ntest)name)abr   c)r   d)r   e)r   f)r   r   g)nxGraphGadd_edges_fromadd_nodeDiGraphDG)clsr   s     ^/home/rose/Desktop/poly/venv/lib/python3.12/site-packages/networkx/readwrite/tests/test_p2g.pysetup_classzTestP2G.setup_class	   sM    f%TQsCEE"    c                     d}t        j                  |      }t        |      }|j                  dk(  sJ t	        |      g dk(  sJ t        |j                         g dd      sJ y )Ns   name
3 4
a
1 2
b

c
0 2
r
   )r   r   r   ))r   r   r   )r   r   )r   r   Tdirected)ioBytesIOr   r
   sortedr   edges)selfsbytesIOr   s       r   test_read_p2gzTestP2G.test_read_p2g   sg    	 **Q-gww&   bz_,,,HHJHSW
 	
 
r   c                     d}t        j                         }t        j                         }d|_        |j                  ddg       t        ||       |j                  d       |j                         }||k(  sJ y )Ns   foo
3 2
1
1 
2
2 
3

foo)      )r-      r   )	r"   r#   r   r   r
   r   r   seekread)r&   r'   fhr   rs        r   test_write_p2gzTestP2G.test_write_p2g$   sf     ZZ\JJL	&&)*!R

GGIAvvr   c                 0   t        j                         }t        j                         }d|_        |j                  ddg       t        ||       |j                  d       t        |      }t        |j                         |j                         d      sJ y )Nr+   r   r   r   Tr    )r"   r#   r   r   r
   r   r   r/   r   r   r%   )r&   r1   r   Hs       r   test_write_read_p2gzTestP2G.test_write_read_p2g7   sn    ZZ\JJL	*j12!R

RL1779aggi$???r   N)__name__
__module____qualname__classmethodr   r)   r3   r6    r   r   r   r      s$    # #
&&@r   r   )	r"   networkxr   networkx.readwrite.p2gr   r   networkx.utilsr   r   r;   r   r   <module>r?      s    	  6 &7@ 7@r   