def test_validate_index_does_not_exist():
bad_examples = [(3, 4, 'template'), (3, 3, 'serve'), (3, -1, 'train')]
good_examples = [(3, 0, 'template'), (3, 1, 'serve'), (3, 2, 'train')]
for n, ind, command in bad_examples:
with pytest.raises(
IndexError,
match=
f"Index {ind} does not exist... Run `kaos {command} list` again"
):
validate_index(n=n, ind=ind, command=command)
for n, ind, command in good_examples:
validate_index(n=n, ind=ind, command=command)
def get_endpoint_by_ind(ind):
data = validate_cache(SERVE_CACHE, command='serve')
loc = validate_index(len(data), ind, command='serve')
return data[loc]['name']
def get_notebook_by_ind(ind):
data = validate_cache(NOTEBOOK_CACHE, command='notebook')
loc = validate_index(len(data), ind, command='notebook')
return data[loc]['name']
def get_job_by_ind(ind):
data = validate_cache(TRAIN_CACHE, command='train')
loc = validate_index(len(data), ind, command='train')
return data[loc]['job_id']
def get_template_name_by_ind(self, ind):
templates = self.list()
loc = validate_index(len(templates), ind, command='template')
return templates[loc]["name"]
def get_workspace_by_ind(ind):
data = validate_cache(WORKSPACE_CACHE, command='workspace')
loc = validate_index(len(data), ind, command='workspace')
return data[loc]['name']
