Return to tree window after test using notebook

takluyver · takluyver · commit 934c7dda8054 · 2019-11-08T16:18:08.000Z
diff --git a/notebook/tests/selenium/conftest.py b/notebook/tests/selenium/conftest.py
@@ -118,4 +118,6 @@ def authenticated_browser(selenium_driver, notebook_server):
 
 @pytest.fixture
 def notebook(authenticated_browser):
-    return Notebook.new_notebook(authenticated_browser)
+    tree_wh = authenticated_browser.current_window_handle
+    yield Notebook.new_notebook(authenticated_browser)
+    authenticated_browser.switch_to.window(tree_wh)
