add test case and remove unused imports
2 files changed