cathoderay
diff --git a/Collapse file
‎test/test_replica_set_connection.py‎
Copy file name to clipboardExpand all lines: test/test_replica_set_connection.py
+2-3Lines changed: 2 additions & 3 deletions b/Collapse file
‎test/test_replica_set_connection.py‎
Copy file name to clipboardExpand all lines: test/test_replica_set_connection.py
+2-3Lines changed: 2 additions & 3 deletions
@@ -32,11 +32,10 @@
 
 from bson.son import SON
 from bson.tz_util import utc
-from pymongo import thread_util
 from pymongo.connection import Connection
 from pymongo.read_preferences import ReadPreference
 from pymongo.replica_set_connection import ReplicaSetConnection
-from pymongo.mongo_replica_set_client import _partition_node
+from pymongo.mongo_replica_set_client import _partition_node, have_gevent
 from pymongo.database import Database
 from pymongo.pool import SocketInfo
 from pymongo.errors import (AutoReconnect,
@@ -194,7 +193,7 @@ def test_use_greenlets(self):
         self.assertFalse(
             ReplicaSetConnection(pair, replicaSet=self.name).use_greenlets)
 
-        if thread_util.have_greenlet:
+        if have_gevent:
             self.assertTrue(ReplicaSetConnection(
                 pair, replicaSet=self.name, use_greenlets=True).use_greenlets)