Fix the test to correctly reflect desired output
Add support for custom import placement to isort