o
    3ήc                     @   s:   d dl Z d dlZd dlZd dlmZmZ G dd dZdS )    N)jit_data	jit_graphc                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )TestJITc                 C   s   t  }|jddd |jddd |d |jdddd	d
 |jddddd
 |dd t|}tt|}t ||sBJ d S )NNode1foobar	node_dataNode3barNode4Node2	   isSomethingweight	something   isNotSomething	nxGraphadd_nodeadd_edger   r   jsonloadsis_isomorphicselfGdK r!   S/tmp/pip-target-vg8gfxp4/lib/python/networkx/readwrite/json_graph/tests/test_jit.pytest_jit
   s   
zTestJIT.test_jitc                 C   s~   t  }|jddd |jddd |jddddd |jddddd |dd t|}tt|}t ||s=J d S )	N      r   r      r   r   r   r   r   r!   r!   r"   
test_jit_2   s   zTestJIT.test_jit_2c                 C   s   t  }|jddd |jddd |jddddd |jddddd |dd t|}tt|t  d	}t ||sAJ d S 
Nr$   r%   r   r   r&   r   r   r   )create_using)	r   DiGraphr   r   r   r   r   r   r   r   r!   r!   r"   test_jit_directed!   s   zTestJIT.test_jit_directedc                 C   s   t  }|jddd |jddd |jddddd |jddddd |dd tt jt| t |}t|}t	t
|t  d	}t ||sNJ |dd t ||r\J t ||sdJ d S r(   )r   MultiDiGraphr   r   pytestraisesNetworkXNotImplementedr   r*   r   r   r   r   )r   r   Hr   r    r!   r!   r"   test_jit_multi_directed,   s   
zTestJIT.test_jit_multi_directedc                 C   s<   t  }t |}tt|}t|}t ||sJ d S )N)r   r   r   r   r   r   r   )r   r   r   r0   r    r!   r!   r"   test_jit_round_trip=   s
   
zTestJIT.test_jit_round_tripN)__name__
__module____qualname__r#   r'   r+   r1   r2   r!   r!   r!   r"   r   	   s    r   )r   r-   networkxr   networkx.readwrite.json_graphr   r   r   r!   r!   r!   r"   <module>   s
    