def test_get_all_siblings_with_siblings(self): node = TreeNode() sibling = TreeNode() node.next_sibling = sibling self.assertEqual([node, sibling], list(node.get_all_siblings()))
def test_get_all_siblings(self): node = TreeNode() self.assertEqual([node], list(node.get_all_siblings()))