refact (Test Main): Fix some PEP violations