diff options
Diffstat (limited to 'src/mango/test/08-text-limit-test.py')
| -rw-r--r-- | src/mango/test/08-text-limit-test.py | 28 |
1 files changed, 14 insertions, 14 deletions
diff --git a/src/mango/test/08-text-limit-test.py b/src/mango/test/08-text-limit-test.py index 4bc87b4b9..ae827813d 100644 --- a/src/mango/test/08-text-limit-test.py +++ b/src/mango/test/08-text-limit-test.py @@ -14,32 +14,32 @@ import mango import limit_docs import unittest + @unittest.skipUnless(mango.has_text_service(), "requires text service") class LimitTests(mango.LimitDocsTextTests): - def test_limit_field(self): - q = {"$or": [{"user_id" : {"$lt" : 10}}, {"filtered_array.[]": 1}]} + q = {"$or": [{"user_id": {"$lt": 10}}, {"filtered_array.[]": 1}]} docs = self.db.find(q, limit=10) assert len(docs) == 8 for d in docs: assert d["user_id"] < 10 def test_limit_field2(self): - q = {"$or": [{"user_id" : {"$lt" : 20}}, {"filtered_array.[]": 1}]} + q = {"$or": [{"user_id": {"$lt": 20}}, {"filtered_array.[]": 1}]} docs = self.db.find(q, limit=10) assert len(docs) == 10 for d in docs: assert d["user_id"] < 20 def test_limit_field3(self): - q = {"$or": [{"user_id" : {"$lt" : 100}}, {"filtered_array.[]": 1}]} + q = {"$or": [{"user_id": {"$lt": 100}}, {"filtered_array.[]": 1}]} docs = self.db.find(q, limit=1) assert len(docs) == 1 for d in docs: assert d["user_id"] < 100 def test_limit_field4(self): - q = {"$or": [{"user_id" : {"$lt" : 0}}, {"filtered_array.[]": 1}]} + q = {"$or": [{"user_id": {"$lt": 0}}, {"filtered_array.[]": 1}]} docs = self.db.find(q, limit=35) assert len(docs) == 0 @@ -52,29 +52,29 @@ class LimitTests(mango.LimitDocsTextTests): assert d["age"] < 100 def test_limit_skip_field1(self): - q = {"$or": [{"user_id" : {"$lt" : 100}}, {"filtered_array.[]": 1}]} + q = {"$or": [{"user_id": {"$lt": 100}}, {"filtered_array.[]": 1}]} docs = self.db.find(q, limit=10, skip=20) assert len(docs) == 10 for d in docs: assert d["user_id"] > 20 def test_limit_skip_field2(self): - q = {"$or": [{"user_id" : {"$lt" : 100}}, {"filtered_array.[]": 1}]} + q = {"$or": [{"user_id": {"$lt": 100}}, {"filtered_array.[]": 1}]} docs = self.db.find(q, limit=100, skip=100) assert len(docs) == 0 def test_limit_skip_field3(self): - q = {"$or": [{"user_id" : {"$lt" : 20}}, {"filtered_array.[]": 1}]} + q = {"$or": [{"user_id": {"$lt": 20}}, {"filtered_array.[]": 1}]} docs = self.db.find(q, limit=1, skip=30) assert len(docs) == 0 def test_limit_skip_field4(self): - q = {"$or": [{"user_id" : {"$lt" : 100}}, {"filtered_array.[]": 1}]} + q = {"$or": [{"user_id": {"$lt": 100}}, {"filtered_array.[]": 1}]} docs = self.db.find(q, limit=0, skip=0) assert len(docs) == 0 def test_limit_skip_field5(self): - q = {"$or": [{"user_id" : {"$lt" : 100}}, {"filtered_array.[]": 1}]} + q = {"$or": [{"user_id": {"$lt": 100}}, {"filtered_array.[]": 1}]} try: self.db.find(q, limit=-1) except Exception as e: @@ -83,7 +83,7 @@ class LimitTests(mango.LimitDocsTextTests): raise AssertionError("Should have thrown error for negative limit") def test_limit_skip_field6(self): - q = {"$or": [{"user_id" : {"$lt" : 100}}, {"filtered_array.[]": 1}]} + q = {"$or": [{"user_id": {"$lt": 100}}, {"filtered_array.[]": 1}]} try: self.db.find(q, skip=-1) except Exception as e: @@ -99,7 +99,6 @@ class LimitTests(mango.LimitDocsTextTests): for i in range(1, len(limit_docs.DOCS), 5): self.run_bookmark_sort_check(i) - def run_bookmark_check(self, size): q = {"age": {"$gt": 0}} seen_docs = set() @@ -121,8 +120,9 @@ class LimitTests(mango.LimitDocsTextTests): bm = None age = 0 while True: - json = self.db.find(q, limit=size, bookmark=bm, sort=["age"], - return_raw=True) + json = self.db.find( + q, limit=size, bookmark=bm, sort=["age"], return_raw=True + ) for doc in json["docs"]: assert doc["_id"] not in seen_docs assert doc["age"] >= age |