master
/ .localenv / lib / python3.5 / site-packages / notebook / tests / selenium / test_prompt_numbers.py

test_prompt_numbers.py @master 

4d078e2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import os
import pytest
import time

# selenium test version for 'prompt_numbers.js' 

def get_prompt(nb, index):
    cell = nb.cells[0]
    return cell.find_element_by_class_name('input').find_element_by_class_name('input_prompt').get_attribute('innerHTML').strip() 

def set_prompt(nb, index, value):
    nb.set_cell_input_prompt(index, value)

def test_prompt_numbers(notebook):
    cell_index = 0
    a = 'print("a")'
    notebook.edit_cell(index=cell_index, content=a)
    assert get_prompt(notebook, cell_index) == "<bdi>In</bdi>&nbsp;[&nbsp;]:"
    set_prompt(notebook, cell_index, 2);
    assert get_prompt(notebook, cell_index) == "<bdi>In</bdi>&nbsp;[2]:"
    set_prompt(notebook, cell_index, 0);
    assert get_prompt(notebook, cell_index) == "<bdi>In</bdi>&nbsp;[0]:"
    set_prompt(notebook, cell_index, "'*'");
    assert get_prompt(notebook, cell_index) == "<bdi>In</bdi>&nbsp;[*]:"
    set_prompt(notebook, cell_index, "undefined");
    assert get_prompt(notebook, cell_index) == "<bdi>In</bdi>&nbsp;[&nbsp;]:"
    set_prompt(notebook, cell_index, "null");
    assert get_prompt(notebook, cell_index) == "<bdi>In</bdi>&nbsp;[&nbsp;]:"