Some fixes on tests

- Fixed wrong named class on test_server.py
- Removed unused imports
- Fixed some PEP8 warnings