Hackerrank Swap Case Python Test Train Split