tests: fixed python formatting