sorting.py

from random import randint
from time import time
from sys import argv

def counting_sort(arr: list) -> list:
  start = time()
  count = [0] * (max(arr)+1)
  result = []
  for i in arr:
    count[i] += 1

  for i in range(max(arr)+1):
    result += [i] * count[i]
  return result, time()-start

def bubble_sort(arr: list) -> list:
  start = time()
  for i in range(len(arr)):
    for j in range(i, len(arr)):
      if arr[i] > arr[j]:
        arr[i], arr[j] = arr[j],arr[i]
  return arr, time()-start

if __name__ == '__main__':
  arg = int(argv[1])
  arr = []

  for i in range(arg):
    arr.append(randint(0,arg))

  bubble, time = bubble_sort(arr)
  print('bubble: {}'.format(time))

  bubble = counting_sort(arr)
  print('counting: {}'.format(time))

  start = time()
  arr.sort()
  print('tim: {}'.format(time() - start))