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)

[((124, 142), 'pathlib.Path', 'Path', (['"""test_nb.py"""'], {}), "('test_nb.py')\n", (128, 142), False, 'from pathlib import Path\n'), ((195, 213), 'nb_helpers.clean.clean_one', 'clean_one', (['TEST_NB'], {}), '(TEST_NB)\n', (204, 213), False, 'from nb_helpers.clean import clean_all, clean_one\n'), ((267, 292), 'nb_helpers.clean.clean_all', 'clean_all', ([], {'path': 'TEST_PATH'}), '(path=TEST_PATH)\n', (276, 292), False, 'from nb_helpers.clean import clean_all, clean_one\n')]