def test_run_notebook_node(): notebook_node = utils.make_notebook_node(["2 + 2"]) utils.run_notebook_node(notebook_node) output = notebook_node.cells[0].outputs[0] assert output.data["text/plain"] == "4"
def test_get_returned_output_text(): notebook_node = utils.make_notebook_node(['2 + 2']) utils.run_notebook_node(notebook_node) output_text = utils.get_output_text(notebook_node.cells[0]) assert "4" in output_text
def test_get_printed_output_text(): notebook_node = utils.make_notebook_node( ['print("Some numbers:")\nfor num in range(6):\n\tprint(num)']) utils.run_notebook_node(notebook_node) output_text = utils.get_output_text(notebook_node.cells[0]) assert "4" in output_text
def test_run_notebook_node_allows_warnings(): cell_input = ("import warnings\n" "warnings.warn(\"deprecated\", DeprecationWarning)") notebook_node = utils.make_notebook_node([cell_input]) utils.run_notebook_node(notebook_node)
def test_run_notebook_node_can_allow_errors(cell_input): notebook_node = utils.make_notebook_node([cell_input]) utils.run_notebook_node(notebook_node, allow_errors=True)
def test_run_notebook_node_disallows_errors_by_default(cell_input): notebook_node = utils.make_notebook_node([cell_input]) with pytest.raises(CellExecutionError): utils.run_notebook_node(notebook_node)