Skip to content

Commit 8b73526

Browse files
authored
Added doctest to randomized_heap.py (TheAlgorithms#11151)
1 parent a13e9c2 commit 8b73526

File tree

1 file changed

+28
-2
lines changed

1 file changed

+28
-2
lines changed

data_structures/heap/randomized_heap.py

+28-2
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,40 @@ def __init__(self, value: T) -> None:
2222

2323
@property
2424
def value(self) -> T:
25-
"""Return the value of the node."""
25+
"""
26+
Return the value of the node.
27+
28+
>>> rhn = RandomizedHeapNode(10)
29+
>>> rhn.value
30+
10
31+
>>> rhn = RandomizedHeapNode(-10)
32+
>>> rhn.value
33+
-10
34+
"""
2635
return self._value
2736

2837
@staticmethod
2938
def merge(
3039
root1: RandomizedHeapNode[T] | None, root2: RandomizedHeapNode[T] | None
3140
) -> RandomizedHeapNode[T] | None:
32-
"""Merge 2 nodes together."""
41+
"""
42+
Merge 2 nodes together.
43+
44+
>>> rhn1 = RandomizedHeapNode(10)
45+
>>> rhn2 = RandomizedHeapNode(20)
46+
>>> RandomizedHeapNode.merge(rhn1, rhn2).value
47+
10
48+
49+
>>> rhn1 = RandomizedHeapNode(20)
50+
>>> rhn2 = RandomizedHeapNode(10)
51+
>>> RandomizedHeapNode.merge(rhn1, rhn2).value
52+
10
53+
54+
>>> rhn1 = RandomizedHeapNode(5)
55+
>>> rhn2 = RandomizedHeapNode(0)
56+
>>> RandomizedHeapNode.merge(rhn1, rhn2).value
57+
0
58+
"""
3359
if not root1:
3460
return root2
3561

0 commit comments

Comments
 (0)