##// END OF EJS Templates
Add test for saving notebook with .py script
Thomas Kluyver -
Show More
@@ -196,6 +196,25 b' class TestNotebookManager(TestCase):'
196 196 self.assertEqual(model['name'], 'Untitled0.ipynb')
197 197 self.assertEqual(model['path'], sub_dir.strip('/'))
198 198
199 def test_save_notebook_with_script(self):
200 with TemporaryDirectory() as td:
201 # Create a notebook
202 nm = FileNotebookManager(notebook_dir=td)
203 nm.save_script = True
204 model = nm.create_notebook_model()
205 name = model['name']
206 path = model['path']
207
208 # Get the model with 'content'
209 full_model = nm.get_notebook_model(name, path)
210
211 # Save the notebook
212 model = nm.save_notebook_model(full_model, name, path)
213
214 # Check that the script was created
215 py_path = os.path.join(td, os.path.splitext(name)[0]+'.py')
216 assert os.path.exists(py_path), py_path
217
199 218 def test_delete_notebook_model(self):
200 219 with TemporaryDirectory() as td:
201 220 # Test in the root directory
General Comments 0
You need to be logged in to leave comments. Login now