o
    i                     @   s@   d dl mZ d dlmZ d dlmZ dd Zdd Zdd	 Zd
S )    )Doc)Graph)Vocabc                  C   s>   t t g dd} t| dd}|jdksJ |j| u sJ d S )Nabcdwordshelloname)r   r   r   r   doc)r   graph r   N/home/ubuntu/.local/lib/python3.10/site-packages/spacy/tests/doc/test_graph.pytest_graph_init   s   r   c                  C   s   t t g dd} t| dd}|d}|d|ksJ |d}t|ddgks,J |j||d	d
d |j||d	ds?J | g ksGJ dd | D dgksUJ dd |	 D dgkscJ dd |	 D g kspJ d S )Nr   r
   r   r   r   )      r   r   oneg      %)labelweight)r   c                 S      g | ]}t |qS r   tuple.0hr   r   r   
<listcomp>       z.test_graph_edges_and_nodes.<locals>.<listcomp>c                 S   r   r   r   r   tr   r   r   r       r!   c                 S   r   r   r   r"   r   r   r   r       r!   )
r   r   r   add_nodeget_nodelistadd_edgehas_edgeheadstails)r   r   node1node2r   r   r   test_graph_edges_and_nodes   s   

r-   c                  C   sd  t t g dd} t| dg dg dd d d}t|j\}}}}dd | D d	gks.J d
d | D dgks<J dd | D d	dgksKJ dd | D g dksZJ dd | D g dksiJ dd | D dd	gksxJ dd | D g dksJ dd | D g ksJ dd | D g ksJ dd | D g dksJ d S )Nr   r
   r   )r   r      r   ))r   r   )r   r0   )r   r   )r   r   )r   nodesedgeslabelsweightsc                 S   r   r   r   r   r   r   r   r    '   r!   z#test_graph_walk.<locals>.<listcomp>r1   c                 S   r   r   r   r   r   r   r   r    (   r!   r   c                 S   r   r   r   r   r   r   r   r    )   r!   c                 S   r   r   r   r   r   r   r   r    *   r!   )r   r1   r   c                 S   r   r   r   r   r   r   r   r    +   r!   c                 S   r   r   r   r   r   r   r   r    ,   r!   c                 S   r   r   r   r"   r   r   r   r    -   r!   )r.   r/   r1   r   c                 S   r   r   r   r"   r   r   r   r    .   r!   c                 S   r   r   r   r"   r   r   r   r    /   r!   c                 S   r   r   r   r"   r   r   r   r    0   r!   )r   r   r   r&   r2   r)   
walk_heads
walk_tails)r   r   node0r+   r,   node3r   r   r   test_graph_walk   s(   "r:   N)	spacy.tokens.docr   spacy.tokens.graphr   spacy.vocabr   r   r-   r:   r   r   r   r   <module>   s    