o
    i                     @   s*   d dl Z d dlmZ G dd de jZdS )    N)v4c                   @   s*   e Zd Zd	ddZdd Zedd ZdS )
NBClientTestsBaseFc                 C   s   t jddddt jdddidt jddd	dt jddd
dt jddddt jddddt jdddidt jdddidg}|rk|t jddg didt jdddddididt jdddidt jdddidg t jdd|dt jddg}t j|dS )z<Build a notebook in memory for use with NotebookClient testsstreamstdouta)nametextdisplay_dataz
text/plainb)datacdstderrefz	image/pngzZw==zapplication/pdfzaA==zapplication/json)         r   r   )r   r   abcg(\.@z$ e $)sourceexecution_countoutputs)r   )cells)nbformat
new_outputextendnew_code_cellnew_markdown_cellnew_notebook)selfwith_json_outputsr   r    r!   N/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/nbclient/tests/base.pybuild_notebook   s.   

z NBClientTestsBase.build_notebookc                 C   s   di iS )z$Build an empty resources dictionary.metadatar!   )r   r!   r!   r"   build_resources'   s   z!NBClientTestsBase.build_resourcesc                 G   s   i }|D ]}| | q|S )N)update)cls	dict_argsoutcomer   r!   r!   r"   merge_dicts+   s   zNBClientTestsBase.merge_dictsN)F)__name__
__module____qualname__r#   r%   classmethodr*   r!   r!   r!   r"   r      s
    
 r   )unittestr   r   TestCaser   r!   r!   r!   r"   <module>   s    