Remove unused imports from Python tests

Summary

We have lots of Python tests with unused imports in our test/suite folder. We could do some cleaning.

Acceptance Criteria (Definition of Done)
No more unused imports.

Suggested Solution
We can use the command that can be found in the following example:

 autoflake --in-place --remove-unused-variables example.py