From 64d4bb8040718f8e5fa30481d0b7409ecb1b0d67 Mon Sep 17 00:00:00 2001
From: David Dougherty <dwdougherty@gmail.com>
Date: Wed, 11 Sep 2024 01:29:54 -0700
Subject: [PATCH] DOC-4194: add TCEs to the range query page (#3374)

---
 doctests/query_range.py | 79 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)
 create mode 100644 doctests/query_range.py

diff --git a/doctests/query_range.py b/doctests/query_range.py
new file mode 100644
index 0000000000..4ef957acfb
--- /dev/null
+++ b/doctests/query_range.py
@@ -0,0 +1,79 @@
+# EXAMPLE: query_range
+# HIDE_START
+import json
+import sys
+import redis
+from redis.commands.json.path import Path
+from redis.commands.search.field import TextField, NumericField, TagField
+from redis.commands.search.indexDefinition import IndexDefinition, IndexType
+from redis.commands.search.query import NumericFilter, Query
+
+r = redis.Redis(decode_responses=True)
+
+# create index
+schema = (
+    TextField("$.description", as_name="description"),
+    NumericField("$.price", as_name="price"),
+    TagField("$.condition", as_name="condition"),
+)
+
+index = r.ft("idx:bicycle")
+index.create_index(
+    schema,
+    definition=IndexDefinition(prefix=["bicycle:"], index_type=IndexType.JSON),
+)
+
+# load data
+with open("data/query_em.json") as f:
+    bicycles = json.load(f)
+
+pipeline = r.pipeline(transaction=False)
+for bid, bicycle in enumerate(bicycles):
+    pipeline.json().set(f'bicycle:{bid}', Path.root_path(), bicycle)
+pipeline.execute()
+# HIDE_END
+
+# STEP_START range1
+res = index.search(Query("@price:[500 1000]"))
+print(res.total)
+# >>> 3
+# REMOVE_START
+assert res.total == 3
+# REMOVE_END
+# STEP_END
+
+# STEP_START range2
+query = Query("*").add_filter(NumericFilter("price", 500, 1000))
+res = index.search(query)
+print(res.total)
+# >>> 3
+# REMOVE_START
+assert res.total == 3
+# REMOVE_END
+# STEP_END
+
+# STEP_START range3
+query = Query("*").add_filter(NumericFilter("price", "(1000", "+inf"))
+res = index.search(query)
+print(res.total)
+# >>> 5
+# REMOVE_START
+assert res.total == 5
+# REMOVE_END
+# STEP_END
+
+# STEP_START range4
+query = Query('@price:[-inf 2000]').sort_by('price').paging(0, 5)
+res = index.search(query)
+print(res.total)
+print(res)
+# >>> Result{7 total, docs: [Document {'id': 'bicycle:0', ... }, Document {'id': 'bicycle:7', ... }, Document {'id': 'bicycle:5', ... }, ...]
+# REMOVE_START
+assert res.total == 7
+# REMOVE_END
+# STEP_END
+
+# REMOVE_START
+# destroy index and data
+r.ft("idx:bicycle").dropindex(delete_documents=True)
+# REMOVE_END