From 2f6477a9e59562310a02f35ba711cfe80348f2b3 Mon Sep 17 00:00:00 2001 From: jpg-130 Date: Wed, 9 Oct 2019 12:13:57 +0530 Subject: [PATCH 1/2] Adding doctests for fractional_knapsack.py --- dynamic_programming/fractional_knapsack.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/dynamic_programming/fractional_knapsack.py b/dynamic_programming/fractional_knapsack.py index 881b6a3969d0..5c4750180ed1 100644 --- a/dynamic_programming/fractional_knapsack.py +++ b/dynamic_programming/fractional_knapsack.py @@ -3,6 +3,10 @@ def fracKnapsack(vl, wt, W, n): + """ + >>> fracKnapsack([60, 100, 120], [10, 20, 30], 50, 3) + 240.0 + """ r = list(sorted(zip(vl, wt), key=lambda x: x[0] / x[1], reverse=True)) vl, wt = [i[0] for i in r], [i[1] for i in r] @@ -16,5 +20,8 @@ def fracKnapsack(vl, wt, W, n): else sum(vl[:k]) ) +if __name__ == "__main__": + import doctest -print("%.0f" % fracKnapsack([60, 100, 120], [10, 20, 30], 50, 3)) + doctest.testmod() +## print("%.0f" % fracKnapsack([60, 100, 120], [10, 20, 30], 50, 3)) From 13becb889edf4b2999bc9a373def89a6f4efbb74 Mon Sep 17 00:00:00 2001 From: John Law Date: Thu, 10 Oct 2019 01:24:26 +0800 Subject: [PATCH 2/2] Update fractional_knapsack.py --- dynamic_programming/fractional_knapsack.py | 1 - 1 file changed, 1 deletion(-) diff --git a/dynamic_programming/fractional_knapsack.py b/dynamic_programming/fractional_knapsack.py index 5c4750180ed1..728cdeb009ac 100644 --- a/dynamic_programming/fractional_knapsack.py +++ b/dynamic_programming/fractional_knapsack.py @@ -24,4 +24,3 @@ def fracKnapsack(vl, wt, W, n): import doctest doctest.testmod() -## print("%.0f" % fracKnapsack([60, 100, 120], [10, 20, 30], 50, 3))