Putting 'user oriented code' inside main condition for having valid imports

Author: alaouimehdi1995

sorts/cycle_sort.py

@@ -50,11 +50,11 @@ def cycle_sort(array):
     except NameError:
         raw_input = input  # Python 3
 
-user_input = raw_input('Enter numbers separated by a comma:\n')
-unsorted = [int(item) for item in user_input.split(',')]
-n = len(unsorted)
-cycle_sort(unsorted)
-
-print("After sort : ")
-for i in range(0, n):
-    print(unsorted[i], end=' ')
+    user_input = raw_input('Enter numbers separated by a comma:\n')
+    unsorted = [int(item) for item in user_input.split(',')]
+    n = len(unsorted)
+    cycle_sort(unsorted)
+
+    print("After sort : ")
+    for i in range(0, n):
+        print(unsorted[i], end=' ')

sorts/pancake_sort.py

@@ -13,4 +13,5 @@ def pancake_sort(arr):
         cur -= 1
     return arr
 
-print(pancake_sort([0,10,15,3,2,9,14,13]))
+if __name__ == '__main__':
+    print(pancake_sort([0,10,15,3,2,9,14,13]))

sorts/topological_sort.py

@@ -28,6 +28,6 @@ def topological_sort(start, visited, sort):
     # return sort
     return sort
 
-
-sort = topological_sort('a', [], [])
-print(sort)
+if __name__ == '__main__':
+    sort = topological_sort('a', [], [])
+    print(sort)

sorts/tree_sort.py

@@ -42,4 +42,5 @@ def tree_sort(arr):
     inorder(root,res)
     return res
 
-print(tree_sort([10,1,3,2,9,14,13]))
+if __name__ == '__main__':
+    print(tree_sort([10,1,3,2,9,14,13]))

sorts/wiggle_sort.py

@@ -9,13 +9,11 @@ def wiggle_sort(nums):
         if (i % 2 == 1) == (nums[i-1] > nums[i]):
             nums[i-1], nums[i] = nums[i], nums[i-1]
 
-
-print("Enter the array elements:\n")
-array=list(map(int,input().split()))
-print("The unsorted array is:\n")
-print(array)
-wiggle_sort(array)
-print("Array after Wiggle sort:\n")
-print(array)
-
-
+if __name__ == '__main__':
+    print("Enter the array elements:\n")
+    array=list(map(int,input().split()))
+    print("The unsorted array is:\n")
+    print(array)
+    wiggle_sort(array)
+    print("Array after Wiggle sort:\n")
+    print(array)