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"
