test_nbmanager.py
34 lines
| 1.1 KiB
| text/x-python
|
PythonLexer
MinRK
|
r7637 | """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 | ||||
Brian E. Granger
|
r10661 | from ..filenbmanager import FileNotebookManager | ||
MinRK
|
r7637 | |||
class TestNotebookManager(TestCase): | ||||
def test_nb_dir(self): | ||||
with TemporaryDirectory() as td: | ||||
Brian Granger
|
r8180 | km = FileNotebookManager(notebook_dir=td) | ||
MinRK
|
r8191 | self.assertEqual(km.notebook_dir, td) | ||
MinRK
|
r7637 | |||
def test_create_nb_dir(self): | ||||
with TemporaryDirectory() as td: | ||||
nbdir = os.path.join(td, 'notebooks') | ||||
Brian Granger
|
r8180 | km = FileNotebookManager(notebook_dir=nbdir) | ||
MinRK
|
r8191 | self.assertEqual(km.notebook_dir, nbdir) | ||
MinRK
|
r7637 | |||
def test_missing_nb_dir(self): | ||||
with TemporaryDirectory() as td: | ||||
nbdir = os.path.join(td, 'notebook', 'dir', 'is', 'missing') | ||||
Brian Granger
|
r8180 | self.assertRaises(TraitError, FileNotebookManager, notebook_dir=nbdir) | ||
MinRK
|
r7637 | |||
def test_invalid_nb_dir(self): | ||||
with NamedTemporaryFile() as tf: | ||||
Brian Granger
|
r8180 | self.assertRaises(TraitError, FileNotebookManager, notebook_dir=tf.name) | ||
MinRK
|
r7637 | |||