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, 10, 6, 28), 'pathlib.Path', 'Path', ({(6, 15, 6, 27): '"""test_nb.py"""'}, {}), "('test_nb.py')", False, 'from pathlib import Path\n'), ((11, 4, 11, 22), 'nb_helpers.clean.clean_one', 'clean_one', ({(11, 14, 11, 21): 'TEST_NB'}, {}), '(TEST_NB)', False, 'from nb_helpers.clean import clean_all, clean_one\n'), ((16, 4, 16, 29), 'nb_helpers.clean.clean_all', 'clean_all', (), '', False, 'from nb_helpers.clean import clean_all, clean_one\n')]