Fix tests

etienneschalk · etienneschalk · commit 1e45e3d5b5f7 · 2024-01-06T16:34:26.000+01:00
diff --git a/datatree/tests/test_treenode.py b/datatree/tests/test_treenode.py
@@ -96,7 +96,7 @@ def test_ancestors(self):
         vito = TreeNode(children={"Michael": michael})
         assert tony.root is vito
         assert tony.parents == (michael, vito)
-        assert tony.ancestors == (vito, michael)
+        assert tony.ancestors == (vito, michael, tony)
 
 
 class TestGetNodes:
diff --git a/datatree/treenode.py b/datatree/treenode.py
@@ -281,7 +281,7 @@ def ancestors(self: Tree) -> Tuple[Tree, ...]:
             "Please use `parents`. Example: `tuple(reversed(node.parents))`",
             DeprecationWarning,
         )
-        return tuple(reversed(self.parents))
+        return tuple((*reversed(self.parents), self))
 
     @property
     def root(self: Tree) -> Tree:

Original file line number	Diff line number	Diff line change
`@@ -281,7 +281,7 @@ def ancestors(self: Tree) -> Tuple[Tree, ...]:`
`281`	`281`	"Please use `parents`. Example: `tuple(reversed(node.parents))`",
`282`	`282`	`DeprecationWarning,`
`283`	`283`	`)`
`284`		`- return tuple(reversed(self.parents))`
	`284`	`+ return tuple((*reversed(self.parents), self))`
`285`	`285`
`286`	`286`	`@property`
`287`	`287`	`def root(self: Tree) -> Tree:`