Cleanup test imports

[elehmer]

Description of PR

This PR removes unused package imports in tests and reorders package order in files.

Summary of changes and related issue

Simply removes unneeded imports.

Relevant motivation and context

[Why did you change this and what applicable context is needed to understand why this change is needed?]

Type of change

• [ ] Bug fix (non-breaking change which fixes an issue)
• [ ] New feature (non-breaking change which adds functionality)
• [ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
• [ ] This change requires a documentation update

Definition of Done Checklist

Practical

• [ ] 80% unit test coverage
• [ ] Documentation
  • [ ] All functions/adjusted functions documented in the readthedocs.
  • [ ] Documentation is pushed
• [ ] Complex code commented
• [ ] Naming conventions followed
  • [ ] Helper functions hidden with _ before the name
• [ ] Context of function is clearly provided
  • [ ] Intent of function is provided
  • [ ] How to test, so that it is not siloed on scientists and anyone can review
  • [ ] Appropriate manual testing was completed
• [ ] Any notebooks known to utilize the affected functions are still working
• [ ] Linting completed and resolved

Conceptual

• [ ] Doesn't replicate existing functionality
• [ ] Aligns with general coding standard of existing functions
• [ ] Matches desired functionality from users/scientists