ó
 ,µ[c           @   sw   d  d l  m Z m Z m Z d  d l Z d  d l Z d  d l Z d  d l Z d  d l	 Td  d l
 Td d d „  ƒ  YZ d S(   iÿÿÿÿ(   t   assert_equalt   assert_raisest   assert_not_equalN(   t   *t   TestP2Gc           B   s,   e  Z d  „  Z d „  Z d „  Z d „  Z RS(   c         C   sf   t  j d d ƒ |  _ d
 d d d d d g } |  j j | ƒ |  j j d	 ƒ t  j |  j ƒ |  _ d  S(   Nt   namet   testt   at   bt   ct   dt   et   ft   g(   R   R   (   R   R	   (   R	   R
   (   R
   R   (   R   R   (   R   R   (   t   nxt   Grapht   Gt   add_edges_fromt   add_nodet   DiGrapht   DG(   t   selfR   (    (    s@   lib/python2.7/site-packages/networkx/readwrite/tests/test_p2g.pyt   setUp   s
    c         C   s§   d } t  j | ƒ } t | ƒ } t | j d ƒ t t | ƒ d d d g ƒ g  | j ƒ  D]$ \ } } t | ƒ t | ƒ f ^ qZ } t | j ƒ  d d d d	 g ƒ d  S(
   Ns   name
3 4
a
1 2
b

c
0 2
R   R   R   R	   (   R   R	   (   R   R   (   R	   R   (   R	   R	   (	   t   iot   BytesIOt   read_p2gR    R   t   sortedt   edgest   strt   assert_edges_equal(   R   t   st   bytesIOR   t   ut   vR   (    (    s@   lib/python2.7/site-packages/networkx/readwrite/tests/test_p2g.pyt   test_read_p2g   s    
7c         C   sq   d } t  j ƒ  } t j ƒ  } d | _ | j d d g ƒ t | | ƒ | j d ƒ | j ƒ  } t	 | | ƒ d  S(	   Ns   foo
3 2
1
1 
2
2 
3

t   fooi   i   i   i    (   i   i   (   i   i   (
   R   R   R   t   OrderedDiGraphR   R   t	   write_p2gt   seekt   readR    (   R   R   t   fhR   t   r(    (    s@   lib/python2.7/site-packages/networkx/readwrite/tests/test_p2g.pyt   test_write_p2g%   s    		c         C   sw   t  j ƒ  } t j ƒ  } d | _ | j d d g ƒ t | | ƒ | j d ƒ t | ƒ } t	 | j
 ƒ  | j
 ƒ  ƒ d  S(   NR#   R   R   R	   i    (   R   R   (   R   R	   (   R   R   R   R   R   R   R%   R&   R   R   R   (   R   R(   R   t   H(    (    s@   lib/python2.7/site-packages/networkx/readwrite/tests/test_p2g.pyt   test_write_read_p2g8   s    	(   t   __name__t
   __module__R   R"   R*   R,   (    (    (    s@   lib/python2.7/site-packages/networkx/readwrite/tests/test_p2g.pyR   
   s   			(    (   t
   nose.toolsR    R   R   t   networkxR   R   t   tempfilet   ost   networkx.readwrite.p2gt   networkx.testingR   (    (    (    s@   lib/python2.7/site-packages/networkx/readwrite/tests/test_p2g.pyt   <module>   s   

