Removed unneeded imports in tests's __init__.py and unified them.