more imports in tests.py (for travis)