o
    3ήc	                     @   s:   d dl mZ d dlZd dlmZ dd ZG dd dZdS )    )eqN)isomorphismc                  C   s^   t g dd gd } | tddddtddddsJ | tddddtddddr-J d S )N)xyz         )isocategorical_node_matchdict)nm r   _/tmp/pip-target-vg8gfxp4/lib/python/networkx/algorithms/isomorphism/tests/test_match_helpers.pytest_categorical_node_match   s   "&r   c                   @   s   e Zd Zdd Zdd ZdS )TestGenericMultiEdgeMatchc                 C   s  t  | _t  | _t  | _t  | _dddd}dddd}dd	d
d}dd	d
d}dddd}dddd}||||||fD ]}| jjdi | q@||||||fD ]}| jjdi | qT||fD ]}| jjdi | qd||fD ]}| jjdi | qtd S )Nedge1r   
   )idminFlowmaxFlowedge2   edge3   u   edge4edge5      edge6r   r	   r         )r   r	   )r	   r   )r   r"   )r"   r#   )nxMultiDiGraphG1G2G3G4add_edge)self
attr_dict1
attr_dict2
attr_dict3
attr_dict4
attr_dict5
attr_dict6	attr_dictr   r   r   setup_method   s>   



z&TestGenericMultiEdgeMatch.setup_methodc                 C   sl  t g dd gd tgd }t ddgd gd tgd }t dd t}t dd t}|| jd d | jd d s=J || jd d | jd d sNJ || jd d | jd d s_J || jd d | jd d spJ || jd d | jd d	 sJ || jd d | jd d	 sJ || jd d | jd d	 rJ || jd d | jd d	 rJ d S )
N)r   flowMinflowMaxr   r4   r5   r	   r   r   r"   r#   )r
   generic_multiedge_matchr   r&   r'   r(   r)   )r+   
full_match
flow_matchmin_flow_matchid_matchr   r   r   test_generic_multiedge_match0   s    """""""&z6TestGenericMultiEdgeMatch.test_generic_multiedge_matchN)__name__
__module____qualname__r3   r;   r   r   r   r   r      s    "r   )	operatorr   networkxr$   networkx.algorithmsr   r
   r   r   r   r   r   r   <module>   s
    