def test_zip_node_leaf():
bt1 = Node(1, Leaf(2), Leaf(3))
bt2 = Leaf(2)
with pytest.raises(AssertionError):
bt1.zip(bt2)
def test_zip_node():
bt1 = Node(1, Leaf(2), Leaf(3))
bt2 = Node(4, Leaf(5), Leaf(6))
exp = Node((1, 4), Leaf((2, 5)), Leaf((3, 6)))
res = bt1.zip(bt2)
assert exp == res