dev: add test case for .tree.yml (file sections)

EvoEvolver · Oct 9, 2023 · 5267e53 · 5267e53
1 parent da07d0f
commit 5267e53
Show file tree

Hide file tree

Showing 11 changed files with 21 additions and 2 deletions.
diff --git a/doc_in_py/core.py b/doc_in_py/core.py
@@ -64,7 +64,7 @@ def extract_module_tree_without_comment(module, root_path):
             member_path = inspect.getfile(member)
         except:
             continue
-        if not member_path.startswith(root_path):
+        if member_path != root_path:
             continue
         true_members.append(member)
         true_member_names.append(name)

diff --git a/doc_in_py/test/testing_module_multi/.tree.yml b/doc_in_py/test/testing_module_multi/.tree.yml
@@ -0,0 +1,10 @@
+sections:
+  first section:
+    - a
+    - b
+  second section:
+    - c
+    - another section:
+      - d
+      - e
+default section: other section
diff --git a/doc_in_py/test/testing_module_multi/__init__.py b/doc_in_py/test/testing_module_multi/__init__.py
@@ -0,0 +1 @@
+from .a import A
diff --git a/doc_in_py/test/testing_module_multi/a.py b/doc_in_py/test/testing_module_multi/a.py
@@ -0,0 +1,2 @@
+class A:
+    pass
diff --git a/doc_in_py/test/testing_module_multi/another.py b/doc_in_py/test/testing_module_multi/another.py
diff --git a/doc_in_py/test/testing_module_multi/b.py b/doc_in_py/test/testing_module_multi/b.py
diff --git a/doc_in_py/test/testing_module_multi/c.py b/doc_in_py/test/testing_module_multi/c.py
diff --git a/doc_in_py/test/testing_module_multi/d.py b/doc_in_py/test/testing_module_multi/d.py
diff --git a/doc_in_py/test/testing_module_multi/e.py b/doc_in_py/test/testing_module_multi/e.py
diff --git a/evonote/.tree.yml b/evonote/.tree.yml
@@ -0,0 +1,6 @@
+sections:
+  core:
+    - notetree
+    - indexing
+    - file_helper
+    - model
diff --git a/evonote/notetree/analysis.py b/evonote/notetree/analysis.py
@@ -27,7 +27,7 @@ def analyze_notetree_sparsity(notetree: Tree):
         print("Content:", note.content)
         print("")
 
-def get_children_heavy_notes(notetree: Tree, min_children=10):
+def get_children_heavy_notes(notetree: Tree, min_children=8):
     """
     Return a list of notes with at least min_children children.
     """