o
    ^i9                     @   s   d Z ddlZddlmZ ddlZddlmZ ddlmZ dgZ	dZ
dd	gd
dZejdeddfddZejddgddd Zejdd ZdS )z+Pytest Fixtures exported by Jupyter Server.    N)Path)AsyncFileContentsManager)AsyncLargeFileManagerzpytest_jupyter.jupyter_serverz(The very model of a modern major generalcatz{connection_file}zTest kernel)argvdisplay_namejp_data_dirreturnc                 C   sv   g d}|D ]2}|  d|}|jdd | d}t }|dkr&dg|d< |t| | d	}|t qd
S )zAConfigures some sample kernelspecs in the Jupyter data directory.)samplesample2badkernelsT)parentszkernel.jsonr   non_existent_pathr   zresource.txtN)joinpathmkdirsample_kernel_jsoncopy
write_textjsondumpssome_resource)r   
spec_namesnamesample_kernel_dirsample_kernel_filekernel_jsonsample_kernel_resources r   W/home/ubuntu/hpml_nyu/venv/lib/python3.10/site-packages/jupyter_server/pytest_plugin.pyjp_kernelspecs   s   


r    TF)paramsc                 C   s   t t|| jdS )z]Returns an AsyncFileContentsManager instance based on the use_atomic_writing parameter value.)root_diruse_atomic_writing)r   strparam)requesttmp_pathr   r   r   jp_contents_manager(   s   r(   c                 C   s   t t| dS )z*Returns an AsyncLargeFileManager instance.)r"   )r   r$   )r'   r   r   r   jp_large_contents_manager.   s   r)   )__doc__r   pathlibr   pytest,jupyter_server.services.contents.filemanagerr   1jupyter_server.services.contents.largefilemanagerr   pytest_pluginsr   r   fixturer    r(   r)   r   r   r   r   <module>   s"    
