PEP8 + removed unused imports + moved to a proper UnitTest library (stdlib's one)