def test_new_notebook(self): with TemporaryDirectory() as td: km = NotebookManager(notebook_dir=td) filename = 'Untitled0.ipynb' filepath = os.path.join(td, filename) notebook_id = km.new_notebook(ndir=td) assert os.path.isfile(filepath) # Now make sure path mapping works assert km.path_mapping[notebook_id] == filepath assert km.find_path(notebook_id) == filepath
def test_delete_notebook(self): with TemporaryDirectory() as td: km = NotebookManager(notebook_dir=td) filename = 'new_test.ipynb' filepath = os.path.join(td, filename) notebook_id = km.new_notebook(ndir=td, name=filename) assert os.path.isfile(filepath) # Now make sure path mapping works assert km.path_mapping[notebook_id] == filepath assert km.find_path(notebook_id) == filepath assert notebook_id in km.mapping assert notebook_id in km.path_mapping assert notebook_id in km.rev_mapping.values() km.delete_notebook(notebook_id) assert notebook_id not in km.mapping assert notebook_id not in km.path_mapping assert notebook_id not in km.rev_mapping.values() assert not os.path.isfile(filepath)