o
    i                     @   s4   d dl mZ d dlmZ dd Zdd Zdd Zd	S )
    )Bynew_markdown_cellc                    s*   ddg  fdd| j D }dd |D S )N	text_cellrenderc                    s0   g | ] t  fd dD r tjdqS )c                    s   g | ]	}|  d v qS )class)get_attribute).0ccell `/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/tests/selenium/test_markdown.py
<listcomp>
   s    z4get_rendered_contents.<locals>.<listcomp>.<listcomp>text_cell_render)allfind_elementr   
CLASS_NAME)r	   clr   r   r      s    z)get_rendered_contents.<locals>.<listcomp>c                 S   s"   g | ]}|d ur| d qS )N	innerHTML)r   strip)r	   xr   r   r   r      s
    )cells)nbrendered_cellsr   r   r   get_rendered_contents   s   
r   c                 C   s*   | dd dD }t |g dksJ d S )Nc                 S   s   g | ]}t |qS r   r   )r	   mdr   r   r   r      s    z&test_markdown_cell.<locals>.<listcomp>)z# Fooz**Bar**z*Baz*z```
x = 1
```z```aaaa
x = 1
```z```python
s = "$"
t = "$"
```)u>   <h1 id="Foo">Foo<a class="anchor-link" href="#Foo">¶</a></h1>z<p><strong>Bar</strong></p>z<p><em>Baz</em></p>z<pre><code>x = 1</code></pre>z@<pre><code class="cm-s-ipython language-aaaa">x = 1</code></pre>a  <pre><code class="cm-s-ipython language-python"><span class="cm-variable">s</span> <span class="cm-operator">=</span> <span class="cm-string">"$"</span>
<span class="cm-variable">t</span> <span class="cm-operator">=</span> <span class="cm-string">"$"</span></code></pre>)r   )prefill_notebookr   r   r   r   test_markdown_cell   s   r   c                 C   sZ   t g d}|D ]"}|   | jd| d}| dd| d ks%J | d qd S )N)                  r!   r    zY
            var cell = IPython.notebook.get_cell(1);
            cell.set_heading_level(z();
            cell.get_text();
        r    # )listadd_markdown_cellbrowserexecute_scriptget_cell_contentsdelete_cell)notebooklsti	cell_textr   r   r   test_markdown_headings!   s   
r2   N)selenium.webdriver.common.byr   nbformat.v4r   r   r   r2   r   r   r   r   <module>   s
    
