implemented global IS_PYMONGO_3

2015-04-10 22:23:56 +02:00
parent c0f1493473
commit 48316ba60d
8 changed files with 32 additions and 21 deletions
--- a/tests/test_replicaset_connection.py
+++ b/tests/test_replicaset_connection.py
@@ -1,11 +1,13 @@
 import sys
+
 sys.path[0:0] = [""]
 import unittest

-import pymongo
 from pymongo import ReadPreference

-if pymongo.version_tuple[0] >= 3:
+from mongoengine.python_support import IS_PYMONGO_3
+
+if IS_PYMONGO_3:
    from pymongo import MongoClient
    CONN_CLASS = MongoClient
    READ_PREF = ReadPreference.SECONDARY