Update a node==None case

joowani · May 22, 2020 · 09f53d6 · 09f53d6
1 parent faae8e2
commit 09f53d6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/binarytree/__init__.py b/binarytree/__init__.py
@@ -344,6 +344,8 @@ def get_parent_node(root, node):
                  /
                 3
     """
+    if node is None:
+        return None
     node_stack = []
     while True:
         if root is not None: