From a112ab6d2a72b1437590e6b9f115efd4e3fb6527 Mon Sep 17 00:00:00 2001 From: M Pacer Date: Thu, 22 Mar 2018 17:44:56 -0700 Subject: [PATCH] add select_kernel function for clicking "new" & selecting a kernel --- notebook/tests/selenium/utils.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/notebook/tests/selenium/utils.py b/notebook/tests/selenium/utils.py index 1aa67d137..9054f7019 100644 --- a/notebook/tests/selenium/utils.py +++ b/notebook/tests/selenium/utils.py @@ -129,6 +129,17 @@ class Notebook: def add_markdown_cell(self, index=-1, content="", render=True): self.add_cell(index, cell_type="markdown") self.edit_cell(index=index, content=content, render=render) + + +def select_kernel(browser, kernel_name='kernel-python3'): + """Clicks the "new" button and selects a kernel from the options. + """ + new_button = wait_for_selector(browser, "#new-buttons", single=True) + new_button.click() + kernel_selector = '#{} a'.format(kernel_name) + kernel = wait_for_selector(browser, kernel_selector, single=True) + kernel.click() + @contextmanager def new_window(browser, selector=None): """Creates new window, switches you to that window, waits for selector if set.