o
     ¢i  ã                   @   s(   d dl Z d dlmZ dd„ Zdd„ ZdS )é    N)Ú	HashEmbedc                  C   sJ   t ddƒ ¡ } |  d¡dksJ ‚|  d¡dksJ ‚|  d¡jdks#J ‚d S )Né@   éè  ÚnVÚnOÚE)r   r   )r   Ú
initializeÚget_dimÚ	get_paramÚshape)Úmodel© r   úV/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/layers/test_hash_embed.pyÚ	test_init   s   r   c                  C   s^   t dddd ¡ } t dddd ¡ }tjddd}|  |¡}| |¡}| ¡ | ¡ ks-J ‚d S )	Nr   r   é   )Úseedé   )r   Úuint64)Údtype)r   r   ÚnumpyÚonesÚpredictÚsum)Úmodel1Úmodel2ÚarrÚvector1Úvector2r   r   r   Útest_seed_changes_bucket   s   

r   )r   Ú	thinc.apir   r   r   r   r   r   r   Ú<module>   s    