get_code has option to get iterative code

kgashok · kgashok · commit e0b5ae368b51 · 2020-08-27T16:57:12.000Z
diff --git a/pygorithm/sorting/merge_sort.py b/pygorithm/sorting/merge_sort.py
@@ -79,11 +79,14 @@ def time_complexities():
     return "Best Case: O(nlogn), Average Case: O(nlogn), Worst Case: O(nlogn)"
 
 
-def get_code():
+def get_code(iter=False):
     """
     easily retrieve the source code
     of the sort function
 
     :return: source code
     """
+    if iter: 
+        return inspect.getsource(sorti) + "\n"
+
     return inspect.getsource(sort) + "\n" + inspect.getsource(merge)
