Skip to content

Commit

Permalink
Adding doctests for sum_of_subset.py (TheAlgorithms#1333)
Browse files Browse the repository at this point in the history
  • Loading branch information
jpg-130 authored and poyea committed Oct 11, 2019
1 parent ea47ae2 commit e678879
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions dynamic_programming/sum_of_subset.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,10 @@
def isSumSubset(arr, arrLen, requiredSum):

"""
>>> isSumSubset([2, 4, 6, 8], 4, 5)
False
>>> isSumSubset([2, 4, 6, 8], 4, 14)
True
"""
# a subset value says 1 if that subset sum can be formed else 0
# initially no subsets can be formed hence False/0
subset = [[False for i in range(requiredSum + 1)] for i in range(arrLen + 1)]
Expand All @@ -22,14 +27,9 @@ def isSumSubset(arr, arrLen, requiredSum):
# uncomment to print the subset
# for i in range(arrLen+1):
# print(subset[i])
print(subset[arrLen][requiredSum])

return subset[arrLen][requiredSum]

if __name__ == "__main__":
import doctest

arr = [2, 4, 6, 8]
requiredSum = 5
arrLen = len(arr)
if isSumSubset(arr, arrLen, requiredSum):
print("Found a subset with required sum")
else:
print("No subset with required sum")
doctest.testmod()

0 comments on commit e678879

Please sign in to comment.