PyTest Integration about testbook HOT 3 CLOSED

@testbook.notebook(path_to_notebook)
def test_notebooks_func(notebook):
  notebook.execute_cell(1)
  notebook.assert_cell_text_output(1, "Hello World")
  notebook.execute_cell(2) # Should use the same kernel without shutting it down until the test completes
  notebook.assert_cell_text_output(2, "Hello World 2")

@pytest.fixture()
def notebook_tester():
  def notebook_loader(nb_path):
    nb = testbook.NotebookClient(nb_path)
    with nb.setup_kernel(**kwargs):
      yield nb
....

from testbook import notebook_tester

def test_notebooks_func(notebook_tester):
  with notebook_tester(nb_path) as notebook:
    notebook.execute_cell(1)
    notebook.assert_cell_text_output(1, "Hello World")
    notebook.execute_cell(2) # Should use the same kernel without shutting it down until the test completes
    notebook.assert_cell_text_output(2, "Hello World 2")

import functools
import inspect

def wrap_test(input):
    parent_scope = inspect.stack()[1][0].f_globals  # Hack alert, needed to setup pytest fixture variables in the same scope as the test :/
    @pytest.fixture(scope='module')
    def added(): # Can make the function name dynamic or whatever is needed
        return input + '_added'

    def wrapper(func):
        @functools.wraps(func)
        def inner_wrapper(*args, **kwargs):
            if 'added' in parent_scope:
                raise RuntimeError(
                    "Can't assign fixture {name} to test because the name is already in use".format(name='added'))
            parent_scope['added'] = added
            try:
                return func(*args, **kwargs) # Can also inspect the function to get the name of the first argument if you want the name assignment to be positional instead of by name
            finally:
                # Cleanup definition if it's still present
                if 'added' in parent_scope:
                    del parent_scope['added']
        return inner_wrapper
    return wrapper

@wrap_test('foo')
def test_something(added):
  print(added)  # 'foo_added'

@testbook.notebook(path_to_notebook)
def test_notebooks_func(notebook):
  notebook.execute_cell(1)
  notebook.assert_cell_text_output(1, "Hello World")
  notebook.execute_cell(2) # Should use the same kernel without shutting it down until the test completes
  notebook.assert_cell_text_output(2, "Hello World 2")