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