fix imports in tests/suite.py