o
    h                     @   sL   d dl Z d dlZd dlT d dlmZmZ e jje jjgZ	dd Z
dd ZdS )    N)*)get_augment_ratioaugment_punctc                  C   s   g d} dd }dd }t | ||dddksJ tt t | || W d    n1 s.w   Y  t | ||ddtd	ksBJ d S )
N)
                        	   
   c                 S      | dkS )Nr    xr   r   _/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/common/test_common_data.py<lambda>       z$test_augment_ratio.<locals>.<lambda>c                 S   r   )Nr   r   r   r   r   r   r      r   g?)desired_ratiog        g?g$I$I?)r   pytestraisesAssertionErrorapprox)datashould_augmentcan_augmentr   r   r   test_augment_ratio	   s   "r   c                  C   s:   g dg} dd }|}t | d||}|ddggksJ d S )N)Simpletest.c                 S   s   | d dkS )Nr!   r   r   r   r   r   r      s    z$test_augment_punct.<locals>.<lambda>g      ?r   r    )r   )r   r   r   new_datar   r   r   test_augment_punct   s
   
r$   )r   stanzastanza.testsstanza.models.common.datar   r   marktravispipeline
pytestmarkr   r$   r   r   r   r   <module>   s    