o
    h                     @   sd   d dl Z d dlZd dlmZmZ e jje jjgZdZ	e j
dddd Zdd	 Zd
d Zdd ZdS )    N)	load_bertextract_bert_embeddingszhf-internal-testing/tiny-bertmodule)scopec                  C   s   t t\} }| |fS )N)r   
BERT_MODEL)mt r	   b/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/common/test_bert_embedding.py	tiny_bert
   s   r   c                 C   s   | \}}dS )z7
    Empty method that just tests loading the bert
    Nr	   )r   r   r   r	   r	   r
   test_load_bert   s   r   c                 C   s2   | \}}t | j}tt||g dg|d d S )N)ThisisatestT)next
parametersdevicer   r   )r   r   r   r   r	   r	   r
   test_run_bert   s   r   c                 C   sr   | \}}t | j}tt||g dg|d}tt||g dg|d}t|dks+J t|d |d s7J d S )N)r   r   -r   r   T)r   r    r   r      r   )r   r   r   r   r   lentorchallclose)r   r   r   r   foobarr	   r	   r
   test_run_bert_empty_word   s   r   )pytestr   #stanza.models.common.bert_embeddingr   r   marktravispipeline
pytestmarkr   fixturer   r   r   r   r	   r	   r	   r
   <module>   s    

