Update test_sort.py