mirror of
https://github.com/jupyter/notebook.git
synced 2025-02-11 12:30:51 +08:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
"""Tests for the notebook manager."""
|
|
|
|
import os
|
|
from unittest import TestCase
|
|
from tempfile import NamedTemporaryFile
|
|
|
|
from IPython.utils.tempdir import TemporaryDirectory
|
|
from IPython.utils.traitlets import TraitError
|
|
|
|
from ..filenbmanager import FileNotebookManager
|
|
|
|
class TestNotebookManager(TestCase):
|
|
|
|
def test_nb_dir(self):
|
|
with TemporaryDirectory() as td:
|
|
km = FileNotebookManager(notebook_dir=td)
|
|
self.assertEqual(km.notebook_dir, td)
|
|
|
|
def test_create_nb_dir(self):
|
|
with TemporaryDirectory() as td:
|
|
nbdir = os.path.join(td, 'notebooks')
|
|
km = FileNotebookManager(notebook_dir=nbdir)
|
|
self.assertEqual(km.notebook_dir, nbdir)
|
|
|
|
def test_missing_nb_dir(self):
|
|
with TemporaryDirectory() as td:
|
|
nbdir = os.path.join(td, 'notebook', 'dir', 'is', 'missing')
|
|
self.assertRaises(TraitError, FileNotebookManager, notebook_dir=nbdir)
|
|
|
|
def test_invalid_nb_dir(self):
|
|
with NamedTemporaryFile() as tf:
|
|
self.assertRaises(TraitError, FileNotebookManager, notebook_dir=tf.name)
|
|
|
|
|