
    ިsg(                     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     T/var/www/html/venv/lib/python3.12/site-packages/networkx/readwrite/tests/test_p2g.pysetup_classzTestP2G.setup_class	   sM    f%TQsCEE"    c                 B   d}t        j                  |      }t        |      }|j                  dk(  sJ t	        |      g dk(  sJ |j                         D cg c]  \  }}t        |      t        |      f }}}t        |j                         g d      sJ y c c}}w )Ns   name
3 4
a
1 2
b

c
0 2
r
   )r   r   r   ))r   r   r   )r   r   )r   r   )ioBytesIOr   r
   sortededgesstrr   )selfsbytesIOr   uvr#   s          r   test_read_p2gzTestP2G.test_read_p2g   s    	 **Q-WvvayO+++./ggi8da#a&#a&!881779&VWWW 9s    B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                 ,   t        j                         }t        j                         }d|_        |j                  ddg       t        ||       |j                  d       t        |      }t        |j                         |j                               sJ y )Nr,   r   r   r   )r    r!   r   r   r
   r   r   r0   r   r   r#   )r%   r2   r   Hs       r   test_write_read_p2gzTestP2G.test_write_read_p2g6   sl    ZZ\JJL	*j12!R

RL1779aggi000r   N)__name__
__module____qualname__classmethodr   r*   r4   r7    r   r   r   r      s$    # #X$&1r   r   )	r    networkxr   networkx.readwrite.p2gr   r   networkx.utilsr   r   r<   r   r   <module>r@      s    	  6 &61 61r   