from pathlib import Path
from nb_helpers.clean import clean_all, clean_one
from tests import TEST_PATH

TEST_PATH
TEST_NB = Path("test_nb.py")


def test_clean_one():
    "clean just one nb"
    clean_one(TEST_NB)


def test_clean_all():
    "clean all test nbs"
    clean_all(path=TEST_PATH)

[(6, 'pathlib.Path', 'Path', False, 'from pathlib import Path\n'), (11, 'nb_helpers.clean.clean_one', 'clean_one', False, 'from nb_helpers.clean import clean_all, clean_one\n'), (16, 'nb_helpers.clean.clean_all', 'clean_all', False, 'from nb_helpers.clean import clean_all, clean_one\n')]