def test_get_assignment_cell_type_default(): """Is the cell type '-' when the assignment metadata is missing?""" cell = NotebookNode() cell.metadata = {} cell_type = util.get_assignment_cell_type(cell) assert cell_type == '-'
def test_get_assignment_cell_type_given(): """Is the cell type correct when the assignment metadata is present?""" cell = NotebookNode() cell.metadata = dict(assignment=dict(cell_type="foo")) cell_type = util.get_assignment_cell_type(cell) assert cell_type == "foo"