o
    h                     @   sX   d dl Z d dlmZ d dlT e jje jjgZdd Zdd Z	dd	 Z
d
d Zdd ZdS )    N)	TreeStack)*c                  C   sZ   t dd dd} | d} | d} g d}|D ]}| j|ks J |  } q| d u s+J d S N      valueparentlength   )r   r   r   )r   pushr   pop)stackexpected_valuesr    r   d/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/constituency/test_tree_stack.pytest_simple	   s   


r   c                  C   sB   t dd dd} | d} | d} t| }t| g dksJ d S r   )r   r   list)r   
stack_listr   r   r   	test_iter   s
   

r   c                  C   s6   t dd dd} | d} | d} t| dksJ d S )Nr   r   r   r   zTreeStack(1, 3, 5))r   r   strr   r   r   r   test_str   s   

r   c                  C   sF   t dd dd} t| dksJ | d} | d} t| dks!J d S )Nr   r   r   r   )r   lenr   r   r   r   r   test_len#   s
   

r   c                  C   s<   t dddd} tddD ]}| |} qt| dksJ dS )zF
    Original stack had a bug where this took exponential time...
    r   Nr   r   (   )r   ranger   r   )r   ir   r   r   test_long_len+   s   r   )pytest%stanza.models.constituency.tree_stackr   stanza.testsmarkpipelinetravis
pytestmarkr   r   r   r   r   r   r   r   r   <module>   s    