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/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)