# Copyright (c) 2023 Graphcore Ltd. All rights reserved.
import pathlib
import nbformat
from nbconvert.preprocessors import ExecutePreprocessor
from examples_utils.testing.test_commands import run_command_fail_explicitly

EXAMPLE_ROOT_DIR = pathlib.Path(__file__).parents[1].resolve()


def test_notebook():
    notebook_filename = EXAMPLE_ROOT_DIR / "Train_TGN.ipynb"
    with open(notebook_filename) as f:
        nb = nbformat.read(f, as_version=4)
    ep = ExecutePreprocessor(timeout=600, kernel_name="python3")
    ep.preprocess(nb, {"metadata": {"path": f"{EXAMPLE_ROOT_DIR}"}})