Remove unused use statements in tests.
Add typehint for test parameters.