python/tests: remove unused imports