1

Author: dkropachev

tests/unit/test_host_connection_pool.py

@@ -22,11 +22,11 @@
 from threading import Thread, Event, Lock
 from unittest.mock import Mock, NonCallableMagicMock, MagicMock
 
-from cassandra.cluster import Session, ShardAwareOptions, NoDelayShardReconnectionScheduler
+from cassandra.cluster import Session, ShardAwareOptions
 from cassandra.connection import Connection
 from cassandra.pool import HostConnection, HostConnectionPool
 from cassandra.pool import Host, NoConnectionsAvailable
-from cassandra.policies import HostDistance, SimpleConvictionPolicy
+from cassandra.policies import HostDistance, SimpleConvictionPolicy, NoDelayShardReconnectionScheduler
 
 LOGGER = logging.getLogger(__name__)