Python unit tests for sorting hw