o
3ήc, @ s d Z ddlZddlZddlZddlZddlmZ G dd dZ G dd dZ
G dd d Z d
d Z d d
Z
dd Zdd Zdd Zdd Zdd Zdd Zdd ZdS ) z*
Tests for VF2 isomorphism algorithm.
N) isomorphismc
@ s e Zd Zddgddgddgddgddgddgddgddgddgddgddgddgg Zd d
gd
d gd d gd d gd
dgddgddgdd
gd d
gd
dgd dgd dgg Zdd Zdd Zdd ZdS )TestWikipediaExampleaghibjcd c C s^ t }t }|| j || j t||}| s J | s&J t |j
}d S )N) nxGraphadd_edges_fromg1edgesg2edgesiso GraphMatcher
is_isomorphicsubgraph_is_monomorphicsortedmappingitems)selfg1g2gmr r$ ]/tmp/pip-target-vg8gfxp4/lib/python/networkx/algorithms/isomorphism/tests/test_isomorphvf2.py
test_graph2 s z TestWikipediaExample.test_graphc C sR t }t }|| j || j |g d}t||}| s'J d S )N)r r
r r ) r r r r r subgraphr r subgraph_is_isomorphic)r r! r" g3r# r$ r$ r%
test_subgraphD s z"TestWikipediaExample.test_subgraphc C sT t }t }|| j |ddgddgddgg t||}| s(J d S )Nr r
r r )r r r r r r r )r r! r" r# r$ r$ r% test_subgraph_monoM s z'TestWikipediaExample.test_subgraph_monoN)__name__
__module__ __qualname__r r r& r* r+ r$ r$ r$ r% r
s< r c @ s( e Zd Zedd Zdd Zdd ZdS )TestVF2GraphDBc C s t | dd}td|dd }t }t|D ]%}td|dd }t|D ]}td|dd }||| q*q| |S )z+Creates a Graph instance from the filename.rb)modez
r; r! _ new_nodesr r" r# r$ r$ r% test_multiedge s
r c C s g d} t td}t t fD ];}|| tdD ]/}t |}t| tt ||}t
||}| s@t
||}nt ||}| sLJ q qd S )N) rd r r
re r r r
r
r
r r r r r
r r
rf r r r rR )rU r6 r r DiGraphr rW rX rY rZ r[ r~ r r r r r r$ r$ r%
test_selfloop s
r c C s g d} | dg }t td}t t fD ]C}|| tdD ]7}t |}t| tt ||}t
||}| t | |
sMt||}nt||}| rYJ q"qd S )N)
rd r re r r r r r rf r r r rR )rU r6 r r r r rW rX rY rZ r[ remove_edges_fromselfloop_edgesr~ r r r r ) edges0r> r; r! r r r r" r# r$ r$ r% test_selfloop_mono s"
r c C s t } t }t }| dd | dd |dd |dd t| |}t| |}t| }t| }ddd|v sEJ ddd|v sNJ ddd|v sWJ ddd|v s`J t|t|ksjJ t|dksrJ d S ) NABCYZr r r r r
)r r r7 r r rU subgraph_isomorphisms_iterrV )r! r" r) gm12gm13xyr$ r$ r% test_isomorphism_iter1 s" r c C s t } t }| dd | dd | dd |dd |dd t| |}t| }dddd|v s r! r" r) gmAgmBr$ r$ r%
test_multipleQ s( (
r c C s t } t }t }t| ||g}t||}| s J | s#J tj| ||gtjd}tj||| gtjd}t||}| sEJ | sKJ d S )N) create_using) objectr r r r r r r r )node1node2node3Gr# Hdgmr$ r$ r% test_noncomparable_nodesp s r c C s t } | d | d | jdddd | jdddd | jdddd t }|d |d |jdddd tj| |td d d
}| sNJ d S ) Nr r
r )labelr r r r r )
edge_match)r r add_noder7 r r categorical_edge_matchr )r SGr# r$ r$ r% test_monomorphism_edge_match s
r )__doc__rB rW r3 networkxr networkx.algorithmsr r r r/ rJ r r r r r r r r r r$ r$ r$ r% s$ I85