Add get_child and some tests for it.

1 files changed, 18 insertions, 0 deletions

diff --git a/src/vytree/vytree.ml b/src/vytree/vytree.ml
index 6a97211..7ee3e65 100644
--- a/src/vytree/vytree.ml
+++ b/src/vytree/vytree.ml
@@ -78,3 +78,21 @@ let rec delete_child node path =
         | Some next_child' -> 
             let new_node = delete_child next_child' names in
             replace_child node new_node
+
+let rec get_child node path =
+    match path with
+    | [] -> raise Empty_path
+    | [name] ->
+        begin
+            let child = find_child node name in
+            match child with
+            | None -> raise Nonexistent_path
+            | Some child' -> child'
+        end
+    | name :: names ->
+        begin
+            let next_child = find_child node name in
+            match next_child with
+            | None -> raise Nonexistent_path
+            | Some child' -> get_child child' names
+        end