31 lines
837 B
Python
31 lines
837 B
Python
|
|
||
|
|
def test_prompt_numbers(prefill_notebook):
|
||
|
|
notebook = prefill_notebook(['print("a")'])
|
||
|
|
|
||
|
|
def get_prompt():
|
||
|
|
return (
|
||
|
|
notebook.cells[0].find_element_by_class_name('input')
|
||
|
|
.find_element_by_class_name('input_prompt')
|
||
|
|
.get_attribute('innerHTML').strip()
|
||
|
|
)
|
||
|
|
|
||
|
|
def set_prompt(value):
|
||
|
|
notebook.set_cell_input_prompt(0, value)
|
||
|
|
|
||
|
|
assert get_prompt() == "<bdi>In</bdi> [ ]:"
|
||
|
|
|
||
|
|
set_prompt(2)
|
||
|
|
assert get_prompt() == "<bdi>In</bdi> [2]:"
|
||
|
|
|
||
|
|
set_prompt(0)
|
||
|
|
assert get_prompt() == "<bdi>In</bdi> [0]:"
|
||
|
|
|
||
|
|
set_prompt("'*'")
|
||
|
|
assert get_prompt() == "<bdi>In</bdi> [*]:"
|
||
|
|
|
||
|
|
set_prompt("undefined")
|
||
|
|
assert get_prompt() == "<bdi>In</bdi> [ ]:"
|
||
|
|
|
||
|
|
set_prompt("null")
|
||
|
|
assert get_prompt() == "<bdi>In</bdi> [ ]:"
|