Update test_insertion_sort.py