Removed import * in tests.

Thanks to flake8 path/to/file.py | awk -F ' ' '{ print $5 }' | sort | uniq