def test_source_bundle2(self):
self.maxDiff = 5000
dir = self.get_dir('pip2')
nb_path = join(dir, 'dummy.ipynb')
# Note that here we are introspecting the environment from within
# the test environment. Don't do this in the production code, which
# runs in the notebook server. We need the introspection to run in
# the kernel environment and not the notebook server environment.
environment = detect_environment(dir)
notebook = self.read_notebook(nb_path)
with make_source_bundle(notebook, environment, dir, extra_files=['data.csv']) as bundle, \
tarfile.open(mode='r:gz', fileobj=bundle) as tar:
names = sorted(tar.getnames())
self.assertEqual(names, [
'data.csv',
'dummy.ipynb',
'manifest.json',
'requirements.txt',
])
reqs = tar.extractfile('requirements.txt').read()
# these are the dependencies declared in our setup.py
self.assertIn(b'notebook', reqs)
self.assertIn(b'nbformat', reqs)
manifest = json.loads(tar.extractfile('manifest.json').read().decode('utf-8'))
# don't check requirements.txt since we don't know the checksum
del manifest['files']['requirements.txt']
# also don't check locale value, just require it be present
del manifest['locale']
del manifest['python']['package_manager']['version']
self.assertEqual(manifest, {
u"version": 1,
u"metadata": {
u"appmode": u"jupyter-static",
u"entrypoint": u"dummy.ipynb"
},
u"python": {
u"version": self.python_version(),
u"package_manager": {
u"name": u"pip",
u"package_file": u"requirements.txt"
}
},
u"files": {
u"dummy.ipynb": {
u"checksum": u"36873800b48ca5ab54760d60ba06703a"
},
u"data.csv": {
u"checksum": u"f2bd77cc2752b3efbb732b761d2aa3c3"
}
}
})
def do_test_html_bundle(self, dir):
self.maxDiff = 5000
nb_path = join(dir, 'dummy.ipynb')
# Note that here we are introspecting the environment from within
# the test environment. Don't do this in the production code, which
# runs in the notebook server. We need the introspection to run in
# the kernel environment and not the notebook server environment.
environment = detect_environment(dir)
notebook = self.read_notebook(nb_path)
# borrowed these from the running notebook server in the container
config_dir = '/home/builder/.jupyter'
config = {
'FileContentsManager': {
'root_dir': '/notebooks'
},
'MappingKernelManager': {
'root_dir': '/notebooks'
},
'NotebookApp': {
'notebook_dir': '/notebooks',
'ip': '*',
'port': 9999,
'answer_yes': True,
'open_browser': False,
'token': '',
'nbserver_extensions': {
'rsconnect_jupyter': True
}
}
}
bundle = make_html_bundle(notebook, "a title", config_dir, dir, config,
logging.getLogger())
tar = tarfile.open(mode='r:gz', fileobj=bundle)
try:
names = sorted(tar.getnames())
self.assertEqual(names, [
'dummy.html',
'manifest.json',
])
manifest = json.loads(
tar.extractfile('manifest.json').read().decode('utf-8'))
self.assertEqual(
manifest, {
u"version": 1,
u"metadata": {
u"appmode": u"static",
u"primary_html": u"dummy.html"
},
})
finally:
tar.close()
bundle.close()
def test_source_bundle1(self):
self.maxDiff = 5000
dir = self.get_dir('pip1')
nb_path = join(dir, 'dummy.ipynb')
# Note that here we are introspecting the environment from within
# the test environment. Don't do this in the production code, which
# runs in the notebook server. We need the introspection to run in
# the kernel environment and not the notebook server environment.
environment = detect_environment(dir)
notebook = self.read_notebook(nb_path)
with make_source_bundle(notebook, environment, dir) as bundle, \
tarfile.open(mode='r:gz', fileobj=bundle) as tar:
names = sorted(tar.getnames())
self.assertEqual(names, [
'dummy.ipynb',
'manifest.json',
'requirements.txt',
])
reqs = tar.extractfile('requirements.txt').read()
self.assertEqual(reqs, b'numpy\npandas\nmatplotlib\n')
manifest = json.loads(
tar.extractfile('manifest.json').read().decode('utf-8'))
# don't check locale value, just require it be present
del manifest['locale']
del manifest['python']['package_manager']['version']
self.assertEqual(
manifest, {
u"version": 1,
u"metadata": {
u"appmode": u"jupyter-static",
u"entrypoint": u"dummy.ipynb"
},
u"python": {
u"version": self.python_version(),
u"package_manager": {
u"name": u"pip",
u"package_file": u"requirements.txt"
}
},
u"files": {
u"dummy.ipynb": {
u"checksum": u"36873800b48ca5ab54760d60ba06703a"
},
u"requirements.txt": {
u"checksum": u"5f2a5e862fe7afe3def4a57bb5cfb214"
}
}
})
def test_file(self):
result = detect_environment(self.get_dir('pip1'))
pip_version = result.pop('pip')
self.assertTrue(version_re.match(pip_version))
locale = result.pop('locale')
self.assertIsInstance(locale, str)
self.assertIn('.', locale)
self.assertEqual(
result, {
'package_manager': 'pip',
'source': 'file',
'filename': 'requirements.txt',
'contents': 'numpy\npandas\nmatplotlib\n',
'python': self.python_version(),
})
def test_pip_freeze(self):
result = detect_environment(self.get_dir('pip2'))
contents = result.pop('contents')
# these are the dependencies declared in our setup.py
self.assertIn('notebook', contents)
self.assertIn('nbformat', contents)
pip_version = result.pop('pip')
self.assertTrue(version_re.match(pip_version))
locale = result.pop('locale')
self.assertIsInstance(locale, str)
self.assertIn('.', locale)
self.assertEqual(
result, {
'package_manager': 'pip',
'source': 'pip_freeze',
'filename': 'requirements.txt',
'python': self.python_version(),
})