fix: sort path for get tree diff

kayjan · Oct 16, 2024 · a82b6d2 · a82b6d2
1 parent d8fea80
commit a82b6d2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/bigtree/tree/helper.py b/bigtree/tree/helper.py
@@ -418,7 +418,7 @@ def get_tree_diff(
             (data_both[indicator_col] != "both")
             | (data_both[path_col].isin(path_changes_deque))
         ]
-    data_both = data_both[[path_col]]
+    data_both = data_both[[path_col]].sort_values(path_col)
     if len(data_both):
         tree_diff = construct.dataframe_to_tree(
             data_both, node_type=tree.__class__, sep=tree.sep

diff --git a/tests/tree/test_helper.py b/tests/tree/test_helper.py
@@ -260,12 +260,12 @@ def test_tree_diff_sep_clash_with_node_name(tree_node):
         tree_only_diff = helper.get_tree_diff(tree_node, other_tree_node)
         expected_str = (
             "a\n"
-            "├── b (-)\n"
-            "│   ├── d (-)\n"
-            "│   └── e (-)\n"
-            "│       ├── g (-)\n"
-            "│       └── h (-)\n"
-            "└── /d (+)\n"
+            "├── /d (+)\n"
+            "└── b (-)\n"
+            "    ├── d (-)\n"
+            "    └── e (-)\n"
+            "        ├── g (-)\n"
+            "        └── h (-)\n"
         )
         assert_print_statement(export.print_tree, expected_str, tree=tree_only_diff)