test_nbconvert_api.py 987 B

12345678910111213141516171819202122232425262728293031
  1. import requests
  2. from notebook.utils import url_path_join
  3. from notebook.tests.launchnotebook import NotebookTestBase
  4. class NbconvertAPI(object):
  5. """Wrapper for nbconvert API calls."""
  6. def __init__(self, request):
  7. self.request = request
  8. def _req(self, verb, path, body=None, params=None):
  9. response = self.request(verb,
  10. url_path_join('api/nbconvert', path),
  11. data=body, params=params,
  12. )
  13. response.raise_for_status()
  14. return response
  15. def list_formats(self):
  16. return self._req('GET', '')
  17. class APITest(NotebookTestBase):
  18. def setUp(self):
  19. self.nbconvert_api = NbconvertAPI(self.request)
  20. def test_list_formats(self):
  21. formats = self.nbconvert_api.list_formats().json()
  22. self.assertIsInstance(formats, dict)
  23. self.assertIn('python', formats)
  24. self.assertIn('html', formats)
  25. self.assertEqual(formats['python']['output_mimetype'], 'text/x-python')