apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-17 15:35:21 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: Replace Tree class with tree()

Browse source
This commit is contained in:
莘权 马 2024-03-12 11:10:09 +08:00
parent e39cafdd58
commit 7d32f9efe6
2 changed files with 149 additions and 77 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/metagpt/utils/test_tree.py
View file

@ -4,7 +4,7 @@ from typing import List
import pytest
from gitignore_parser import parse_gitignore
from metagpt.utils.tree import Tree
from metagpt.utils.tree import _print_tree, tree
@pytest.mark.parametrize(
@ -16,8 +16,8 @@ from metagpt.utils.tree import Tree
)
def test_tree(root: str, rules: str):
gitignore_rules = parse_gitignore(full_path=rules) if rules else None
tree = Tree(root=root).print(git_ignore_rules=gitignore_rules)
assert tree
v = tree(root=root, git_ignore_rules=gitignore_rules)
assert v
@pytest.mark.parametrize(
@ -46,7 +46,7 @@ def test_tree(root: str, rules: str):
],
)
def test__print_tree(tree: dict, want: List[str]):
v = Tree._print_tree(tree)
v = _print_tree(tree)
assert v == want