cătălin
ab870b5910
Some checks reported errors
continuous-integration/drone Build encountered an error
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
import pytest
|
|
|
|
from halig.commands.notebooks import NotebooksCommand
|
|
|
|
|
|
@pytest.mark.usefixtures('_notes')
|
|
def test_build_tree_max_depth_0(notebooks_command: NotebooksCommand):
|
|
notebooks_command.max_depth = 0
|
|
tree = notebooks_command.build_tree(notebooks_command.settings.notebooks_root_path)
|
|
assert not tree.children
|
|
|
|
|
|
@pytest.mark.usefixtures('_notes')
|
|
def test_build_tree_max_depth_1(notebooks_command: NotebooksCommand):
|
|
notebooks_command.max_depth = 1
|
|
tree = notebooks_command.build_tree(notebooks_command.settings.notebooks_root_path)
|
|
personal = tree.children[0]
|
|
work = tree.children[1]
|
|
assert personal.label == "Personal"
|
|
assert work.label == "Work"
|
|
assert not personal.children
|
|
assert not work.children
|
|
|
|
|
|
@pytest.mark.usefixtures('_notes')
|
|
def test_build_tree_max_depth_2(notebooks_command: NotebooksCommand):
|
|
notebooks_command.max_depth = 2
|
|
tree = notebooks_command.build_tree(notebooks_command.settings.notebooks_root_path)
|
|
personal = tree.children[0]
|
|
work = tree.children[1]
|
|
assert personal.label == "Personal"
|
|
assert work.label == "Work"
|
|
assert len(work.children) == 2 # noqa: PLR2004
|
|
assert len(personal.children) == 1
|
|
|
|
|
|
@pytest.mark.usefixtures('_notes')
|
|
def test_build_tree_max_depth_inf(notebooks_command: NotebooksCommand):
|
|
tree = notebooks_command.build_tree(notebooks_command.settings.notebooks_root_path)
|
|
personal = tree.children[0]
|
|
work = tree.children[1]
|
|
assert personal.label == "Personal"
|
|
assert work.label == "Work"
|
|
assert len(work.children) == 2 # noqa: PLR2004
|
|
assert len(personal.children) == 1
|
|
|
|
assert work.children[0].label == "Dailies"
|
|
assert len(work.children[0].children) == 10 # noqa: PLR2004
|