remove unused import from test, thanks flake8
F403: do not use import *