a
    =b                     @   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   >lib/python3.9/site-packages/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sfJ |d |ksvJ |d d dksJ |d d dksJ d S )Nzhello worldzf = lambda x: x * xZmarkdown)	cell_typecodeZ
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   Zadd_cellr   r   )r   Zstr1Zstr2dr   valuesr   r   r	   test_notebook_add   s    r   )r   Znotebook_docr   r   r
   r   r   r   r   r	   <module>   s   