Sum binary tree path ints

2023-10-21 16:34:42 -04:00
parent e92d117a41
commit cee338520e
2 changed files with 45 additions and 0 deletions
@@ -177,3 +177,20 @@ def test_connect_binary_tree_right(
    connected, serialized = stuff.connect_binary_tree_right(root)
    assert connected is not None
    assert serialized == expected
+
+
+@pytest.mark.parametrize(
+    ("root", "expected"),
+    [
+        (
+            stdlib.Node(
+                4,
+                right=stdlib.Node(0),
+                left=stdlib.Node(9, right=stdlib.Node(1), left=stdlib.Node(5)),
+            ),
+            1026,
+        ),
+    ],
+)
+def test_connect_binary_tree_sum_numbers(root: stdlib.Node | None, expected: int):
+    assert stuff.sum_binary_tree_path_ints(root) == expected