You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
987 B
31 lines
987 B
import requests
|
|
|
|
from notebook.utils import url_path_join
|
|
from notebook.tests.launchnotebook import NotebookTestBase
|
|
|
|
class NbconvertAPI(object):
|
|
"""Wrapper for nbconvert API calls."""
|
|
def __init__(self, request):
|
|
self.request = request
|
|
|
|
def _req(self, verb, path, body=None, params=None):
|
|
response = self.request(verb,
|
|
url_path_join('api/nbconvert', path),
|
|
data=body, params=params,
|
|
)
|
|
response.raise_for_status()
|
|
return response
|
|
|
|
def list_formats(self):
|
|
return self._req('GET', '')
|
|
|
|
class APITest(NotebookTestBase):
|
|
def setUp(self):
|
|
self.nbconvert_api = NbconvertAPI(self.request)
|
|
|
|
def test_list_formats(self):
|
|
formats = self.nbconvert_api.list_formats().json()
|
|
self.assertIsInstance(formats, dict)
|
|
self.assertIn('python', formats)
|
|
self.assertIn('html', formats)
|
|
self.assertEqual(formats['python']['output_mimetype'], 'text/x-python') |