Skip to content

Commit ddb0949

Browse files
jpg-130cclauss
authored andcommitted
Adding doctests for fractional_knapsack.py (#1331)
* Adding doctests for fractional_knapsack.py * Update fractional_knapsack.py
1 parent 9c63473 commit ddb0949

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

dynamic_programming/fractional_knapsack.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,10 @@
33

44

55
def fracKnapsack(vl, wt, W, n):
6+
"""
7+
>>> fracKnapsack([60, 100, 120], [10, 20, 30], 50, 3)
8+
240.0
9+
"""
610

711
r = list(sorted(zip(vl, wt), key=lambda x: x[0] / x[1], reverse=True))
812
vl, wt = [i[0] for i in r], [i[1] for i in r]
@@ -16,5 +20,7 @@ def fracKnapsack(vl, wt, W, n):
1620
else sum(vl[:k])
1721
)
1822

23+
if __name__ == "__main__":
24+
import doctest
1925

20-
print("%.0f" % fracKnapsack([60, 100, 120], [10, 20, 30], 50, 3))
26+
doctest.testmod()

0 commit comments

Comments
 (0)