o
    h                     @   sh   d dl Z d dlZd dlmZ d dlmZ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S )    N)Pipeline)SinusoidalEncodingAddSinusoidalEncoding)*c                  C   .   t ddd} | tdg}|jdksJ d S )N
      	model_dimmax_len      r   r   torchtensorshapeencodingfoo r   m/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/constituency/test_positional_encoding.pytest_positional_encoding      r   c                  C   r   )Nr      r	   r   r   r   r   r   r   r   test_resize   r   r   c                  C   s<   t ddd} | td}|jdksJ |  dksJ d S )Nr      r	      )r   r   )r   r   aranger   r   r   r   r   r   test_arange   s   r   c                  C   s   t ddd} tddd}| |}tddd}| |}tj|| |dds(J tddd}| |}tj|d |d  |ddsCJ tj|d |d  |ddsSJ d S )	Nr   r   )d_modelr   r   gHz>)atolr   r   )r   r   zerosrandnallclose)r   xyrr2r   r   r   test_add   s    $r)   )pytestr   stanzar   .stanza.models.constituency.positional_encodingr   r   stanza.testsmarkpipelinetravis
pytestmarkr   r   r   r)   r   r   r   r   <module>   s    