def test_get_rcdata():
with temp_directory():
expected_result = {"notebooks": {}}
assert get_rcdata() == expected_result
with fake_rc():
assert get_rcdata() == _data
def test_get_rcdata_rcfile_exists(self):
with open(RC_FILENAME, 'w') as f:
json.dump(data, f, indent=2)
expected_result = data
self.assertEqual(get_rcdata(), expected_result)
def test_set_notebook_slug_existing_entry(self):
filename = "Testing Jovian.ipynb"
slug = "46bd9a3f87e74de0baf8a6f0b60a8df9"
expected_result = data
set_notebook_slug(filename, slug)
self.assertEqual(get_rcdata(), expected_result)
def pull(slug=None):
"""Get the latest files associated with the current gist"""
# If a slug is provided, just use that
if slug:
clone(slug, fresh=False)
return
# Check if .jovianrc exists
if not rcfile_exists():
log(RCFILE_NOTFOUND, error=True)
return
# Get list of notebooks
nbs = get_rcdata()['notebooks']
for fname in nbs:
# Get the latest files for each notebook
clone(nbs[fname]['slug'], fresh=False)
def test_set_notebook_slug():
with fake_rc():
filename = "Testing Jovian 2.ipynb"
slug = "46bd9a3f87e74de0baf8a6f0b60a8df9"
expected_result = {
"notebooks": {
"Testing Jovian.ipynb": {
"slug": "46bd9a3f87e74de0baf8a6f0b60a8df9"
},
"Testing Jovian 2.ipynb": {
"slug": "46bd9a3f87e74de0baf8a6f0b60a8df9"
}
}
}
set_notebook_slug(filename, slug)
assert get_rcdata() == expected_result
def test_set_notebook_slug_new_entry(self):
filename = "Testing Jovian 2.ipynb"
slug = "46bd9a3f87e74de0baf8a6f0b60a8df9"
expected_result = {
"notebooks": {
"Testing Jovian.ipynb": {
"slug": "46bd9a3f87e74de0baf8a6f0b60a8df9"
},
"Testing Jovian 2.ipynb": {
"slug": "46bd9a3f87e74de0baf8a6f0b60a8df9"
}
}
}
set_notebook_slug(filename, slug)
self.assertEqual(get_rcdata(), expected_result)
def test_get_rcdata_rcfile_does_not_exist(self):
expected_result = {
"notebooks": {}
}
self.assertEqual(get_rcdata(), expected_result)
