o
    i                     @   sH   d dl mZ d dlmZ ddlmZ G dd deZdd Zd	d
 ZdS )    )List)Shim   )make_tempdirc                       s@   e Zd Zdee f fddZdd Zdedd fddZ  Z	S )	MockShimdatac                    s   t  jd d d d || _d S )N)config	optimizer)super__init__r   selfr   	__class__ P/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/layers/test_shim.pyr   	   s   
zMockShim.__init__c                 C   s
   t | jS )N)bytesr   )r   r   r   r   to_bytes   s   
zMockShim.to_bytesreturnc                 C   s   t t|dS )N)r   )r   listr   r   r   r   
from_bytes   s   zMockShim.from_bytes)
__name__
__module____qualname__r   intr   r   r   r   __classcell__r   r   r   r   r      s    r   c                  C   sf   t  } | d }tg d}|| ||}W d    n1 s"w   Y  | | ks1J d S Nzcool_shim.data)   r      )r   r   to_disk	from_diskr   )path	shim_pathshim	copy_shimr   r   r   !test_shim_can_roundtrip_with_path   s   
r%   c                 C   s@   | d }t g d}|| ||}| | ksJ d S r   )r   r   r    r   )pathy_fixturer"   r#   r$   r   r   r   *test_shim_can_roundtrip_with_path_subclass   s
   

r'   N)	typingr   thinc.shims.shimr   utilr   r   r%   r'   r   r   r   r   <module>   s    
