o
    ?Hh                     @   st   d dl Z d dlmZmZ d dlmZ d dlmZ d dlm	Z	m
Z
 edd Zeddd	ged
ddgdd ZdS )    N)make_memmapconcurrency_safe_rename)
with_numpy)parametrize)Paralleldelayedc                 C   sB   |  dj}dtj }tjd }t||d|d}|j|ksJ d S )Nz	test.mmap      zw+)shapemodeoffset)joinstrpathmmapALLOCATIONGRANULARITYr   r   )tmpdirfnamesizer   
memmap_obj r   Z/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/joblib/test/test_backports.pytest_memmap	   s
   

r   dst_contentzdst contentbackend	threadingc                    s   fddt dD }|D ]}|d qd |d ur# | td|d fdd|D    s8J   dks@J |D ]}| rJJ qBd S )	Nc                    s   g | ]	}  d | qS )zsrc_%d)r   ).0i)r   r   r   
<listcomp>   s    z0test_concurrency_safe_rename.<locals>.<listcomp>   zsrc contentdst)n_jobsr   c                 3   s"    | ]}t t|j jV  qd S )N)r   r   r   )r   src_path)dst_pathr   r   	<genexpr>   s
    
z/test_concurrency_safe_rename.<locals>.<genexpr>)rangewriter   r   existsread)r   r   r   	src_pathsr!   r   )r"   r   r   test_concurrency_safe_rename   s   

r)   )r   joblib.backportsr   r   joblib.test.commonr   joblib.testingr   joblibr   r   r   r)   r   r   r   r   <module>   s    
