Add more module imports in test.py.