Show More
@@ -1,34 +1,52 | |||
|
1 | 1 | """Tests for the notebook manager.""" |
|
2 | 2 | |
|
3 | 3 | import os |
|
4 | 4 | from unittest import TestCase |
|
5 | 5 | from tempfile import NamedTemporaryFile |
|
6 | 6 | |
|
7 | 7 | from IPython.utils.tempdir import TemporaryDirectory |
|
8 | 8 | from IPython.utils.traitlets import TraitError |
|
9 | 9 | |
|
10 | 10 | from ..filenbmanager import FileNotebookManager |
|
11 | from ..nbmanager import NotebookManager | |
|
11 | 12 | |
|
12 | class TestNotebookManager(TestCase): | |
|
13 | class TestFileNotebookManager(TestCase): | |
|
13 | 14 | |
|
14 | 15 | def test_nb_dir(self): |
|
15 | 16 | with TemporaryDirectory() as td: |
|
16 | 17 | km = FileNotebookManager(notebook_dir=td) |
|
17 | 18 | self.assertEqual(km.notebook_dir, td) |
|
18 | 19 | |
|
19 | 20 | def test_create_nb_dir(self): |
|
20 | 21 | with TemporaryDirectory() as td: |
|
21 | 22 | nbdir = os.path.join(td, 'notebooks') |
|
22 | 23 | km = FileNotebookManager(notebook_dir=nbdir) |
|
23 | 24 | self.assertEqual(km.notebook_dir, nbdir) |
|
24 | 25 | |
|
25 | 26 | def test_missing_nb_dir(self): |
|
26 | 27 | with TemporaryDirectory() as td: |
|
27 | 28 | nbdir = os.path.join(td, 'notebook', 'dir', 'is', 'missing') |
|
28 | 29 | self.assertRaises(TraitError, FileNotebookManager, notebook_dir=nbdir) |
|
29 | 30 | |
|
30 | 31 | def test_invalid_nb_dir(self): |
|
31 | 32 | with NamedTemporaryFile() as tf: |
|
32 | 33 | self.assertRaises(TraitError, FileNotebookManager, notebook_dir=tf.name) |
|
33 | 34 | |
|
35 | class TestNotebookManager(TestCase): | |
|
36 | def test_named_notebook_path(self): | |
|
37 | nm = NotebookManager() | |
|
38 | ||
|
39 | # doesn't end with ipynb, should just be path | |
|
40 | name, path = nm.named_notebook_path('hello') | |
|
41 | self.assertEqual(name, None) | |
|
42 | self.assertEqual(path, 'hello/') | |
|
43 | ||
|
44 | name, path = nm.named_notebook_path('hello.ipynb') | |
|
45 | self.assertEqual(name, 'hello.ipynb') | |
|
46 | self.assertEqual(path, None) | |
|
47 | ||
|
48 | name, path = nm.named_notebook_path('/this/is/a/path/hello.ipynb') | |
|
49 | self.assertEqual(name, 'hello.ipynb') | |
|
50 | self.assertEqual(path, '/this/is/a/path/') | |
|
51 | ||
|
34 | 52 |
General Comments 0
You need to be logged in to leave comments.
Login now