o
    i[
                     @   sf   d dl Z d dlmZ d dlmZ d dlmZ e jdd Zdd Z	d	d
 Z
dd Zdd Zdd ZdS )    N)util)DocVocabc                   C   s   t  S )Nr    r   r   Q/home/ubuntu/.local/lib/python3.10/site-packages/spacy/tests/doc/test_creation.pyvocab   s   r   c                 C   s   t | }t|dksJ d S )Nr   )r   lenr   docr   r   r   test_empty_doc   s   r   c                 C   s@   t | dgd}|jdksJ t | dgdgd}|jdksJ d S )Na)wordsza Fr   spaces)r   textr
   r   r   r   test_single_word   s   r   c                 C   s  g d}d}t ||\}}t| ||d}dd |D g dks"J dd |D g dks/J |j|ks6J d	d |D d
d |D ksFJ g d}d}t ||\}}t| ||d}dd |D g dkshJ dd |D g dksuJ |j|ks|J dd |D dd |D ksJ g d}d}t ||\}}t| ||d}dd |D g dksJ dd |D g dksJ |j|ksJ dd |D dd |D ksJ tt g d}d}t |dg |\}}W d    d S 1 sw   Y  d S )N)'dogsr   runz  'dogs'

run  r   c                 S      g | ]}|j qS r   r   .0tr   r   r   
<listcomp>       z3test_create_from_words_and_text.<locals>.<listcomp>)z  r   r   r   

r    c                 S   r   r   whitespace_r   r   r   r   r       r   ) r!   r!   r!   r!   r   r!   c                 S      g | ]
}|j  s|j qS r   r   isspacer   r   r   r   r   "       c                 S      g | ]}|  s|qS r   r$   r   wordr   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   c                 S   r"   r   r#   r   r   r   r   r   .   r%   c                 S   r&   r   r'   r(   r   r   r   r   .   r*   )r   r   r   r   r   r   r   c                 S   r   r   r   r   r   r   r   r   7   r   c                 S   r   r   r   r   r   r   r   r   8   r   c                 S   r"   r   r#   r   r   r   r   r   :   r%   c                 S   r&   r   r'   r(   r   r   r   r   :   r*   away)r   get_words_and_spacesr   r   pytestraises
ValueError)r   r   r   r   r   r   r   r   test_create_from_words_and_text   sF   "r0   c                 C   sV   d  }ttt|}tt t| ||d W d    d S 1 s$w   Y  d S )NI like ginger)r   heads)splitlistranger	   r-   r.   r/   r   )r   r   r2   r   r   r   "test_create_with_heads_and_no_depsE   s
   "r6   c                 C   sN   d  }d  }tt t| ||d W d    d S 1 s w   Y  d S )Nr1   zQQ ZZ XX)r   pos)r3   r-   r.   r/   r   )r   r   r7   r   r   r   test_create_invalid_posL   s
   "r8   )r-   spacyr   spacy.tokensr   spacy.vocabr   fixturer   r   r   r0   r6   r8   r   r   r   r   <module>   s    
,