Skip to content

Commit

Permalink
Adding missing Doctests (TheAlgorithms#1330)
Browse files Browse the repository at this point in the history
* Adding doctests in abbreviation

* Adding doctests in fibonacci.py
  • Loading branch information
jpg-130 authored and cclauss committed Oct 9, 2019
1 parent b6cc37d commit ea47ae2
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 2 deletions.
12 changes: 10 additions & 2 deletions dynamic_programming/abbreviation.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,13 @@
daBcd -> capitalize a and c(dABCd) -> remove d (ABC)
"""


def abbr(a, b):
"""
>>> abbr("daBcd", "ABC")
True
>>> abbr("dBcd", "ABC")
False
"""
n = len(a)
m = len(b)
dp = [[False for _ in range(m + 1)] for _ in range(n + 1)]
Expand All @@ -28,4 +33,7 @@ def abbr(a, b):


if __name__ == "__main__":
print(abbr("daBcd", "ABC")) # expect True
# print(abbr("daBcd", "ABC")) # expect True
import doctest

doctest.testmod()
16 changes: 16 additions & 0 deletions dynamic_programming/fibonacci.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,20 @@ def __init__(self, N=None):
self.fib_array.append(self.fib_array[i - 1] + self.fib_array[i - 2])
elif N == 0:
self.fib_array.append(0)
print(self.fib_array)

def get(self, sequence_no=None):
"""
>>> Fibonacci(5).get(3)
[0, 1, 1, 2, 3, 5]
[0, 1, 1, 2]
>>> Fibonacci(5).get(6)
[0, 1, 1, 2, 3, 5]
Out of bound.
>>> Fibonacci(5).get(-1)
[0, 1, 1, 2, 3, 5]
[]
"""
if sequence_no != None:
if sequence_no < len(self.fib_array):
return print(self.fib_array[: sequence_no + 1])
Expand Down Expand Up @@ -46,3 +58,7 @@ def get(self, sequence_no=None):
print("\nInvalid input, please try again.")
except NameError:
print("\n********* Invalid input, good bye!! ************\n")

import doctest

doctest.testmod()

0 comments on commit ea47ae2

Please sign in to comment.