o
    i                  	   @   sx   d dl Z d dlZd dlZd dlmZ d dlZd dlmZ e jdd Z	ddddej
d	ej
d
edefddZdd ZdS )    N)Path)NumpyOpsc                  c   s&    t t } | V  tt|  d S N)r   tempfilemkdtempshutilrmtreestr)d r   Y/home/ubuntu/.local/lib/python3.10/site-packages/spacy_curated_transformers/tests/util.pymake_tempdir
   s   r   gh㈵>atolrtolabr   r   c                C   s   t jj| |||d d S )Nr   )torchtestingassert_close)r   r   r   r   r   r   r   torch_assertclose   s   
r   c                 C   sZ   t | ts| g} t |ts|g}t }t| |D ]\}}|jj|||| qd S r   )
isinstancelistr   zipxpr   assert_array_equalasarray)r   r   opsxyr   r   r   xp_assert_array_equal   s   

r    )
contextlibr   r   pathlibr   r   	thinc.apir   contextmanagerr   Tensorfloatr   r    r   r   r   r   <module>   s&    

