o
    :ήc                     @   s,   d dl Z ddlmZmZ dd Zdd ZdS )    N   )Notebookskeleton_nbc                  C   s&   t  } t|  ttksJ d S )N)r   jsonloadsr   )nb r   F/tmp/pip-target-vg8gfxp4/lib/python/doc/ext/tests/test_notebook_doc.pytest_notebook_basic   s    r
   c                  C   s   t  } d}d}| j|dd | j|dd t|  }|d d d }d	d
 |D }|d |ks3J |d |ks;J |d d dksEJ |d d dksOJ d S )Nzhello worldzf = lambda x: x * xmarkdown)	cell_typecode
worksheetsr   cellsc                 S   s(   g | ]}|d  dkr|d n|d qS )r   r   inputsourcer   ).0cr   r   r	   
<listcomp>   s     z%test_notebook_add.<locals>.<listcomp>   r   r   )r   add_cellr   r   )r   str1str2dr   valuesr   r   r	   test_notebook_add   s   r   )r   notebook_docr   r   r
   r   r   r   r   r	   <module>   s    