diff --git a/test/aaa_profiling/test_zoomark.py b/test/aaa_profiling/test_zoomark.py index 7a9a0e11d3..f4ffb7c627 100644 --- a/test/aaa_profiling/test_zoomark.py +++ b/test/aaa_profiling/test_zoomark.py @@ -7,7 +7,8 @@ import datetime import sys import time from sqlalchemy import * -from sqlalchemy.testing import * +from sqlalchemy.testing import fixtures, engines, profiling +from sqlalchemy import testing ITERATIONS = 1 dbapi_session = engines.ReplayableSession() metadata = None diff --git a/test/aaa_profiling/test_zoomark_orm.py b/test/aaa_profiling/test_zoomark_orm.py index ef60960220..78631b8fab 100644 --- a/test/aaa_profiling/test_zoomark_orm.py +++ b/test/aaa_profiling/test_zoomark_orm.py @@ -8,7 +8,8 @@ import sys import time from sqlalchemy import * from sqlalchemy.orm import * -from sqlalchemy.testing import * +from sqlalchemy.testing import fixtures, engines, profiling +from sqlalchemy import testing ITERATIONS = 1 dbapi_session = engines.ReplayableSession() metadata = None