|
27 | 27 |
|
28 | 28 | r.delete("user:1", "user:2", "user:3")
|
29 | 29 | # REMOVE_END
|
30 |
| - |
31 | 30 | # STEP_START create_data
|
32 | 31 | user1 = {
|
33 | 32 | "name": "Paul John",
|
|
65 | 64 | )
|
66 | 65 | )
|
67 | 66 | # STEP_END
|
68 |
| - |
69 | 67 | # Tests for 'make_index' step.
|
70 | 68 | # REMOVE_START
|
71 | 69 | assert indexCreated
|
72 | 70 | # REMOVE_END
|
73 | 71 |
|
74 |
| - |
75 | 72 | # STEP_START add_data
|
76 | 73 | user1Set = r.json().set("user:1", Path.root_path(), user1)
|
77 | 74 | user2Set = r.json().set("user:2", Path.root_path(), user2)
|
78 | 75 | user3Set = r.json().set("user:3", Path.root_path(), user3)
|
79 | 76 | # STEP_END
|
80 |
| - |
81 | 77 | # Tests for 'add_data' step.
|
82 | 78 | # REMOVE_START
|
83 | 79 | assert user1Set
|
84 | 80 | assert user2Set
|
85 | 81 | assert user3Set
|
86 | 82 | # REMOVE_END
|
87 | 83 |
|
88 |
| - |
89 | 84 | # STEP_START query1
|
90 | 85 | findPaulResult = r.ft("idx:users").search(
|
91 | 86 | Query("Paul @age:[30 40]")
|
|
94 | 89 | print(findPaulResult)
|
95 | 90 | # >>> Result{1 total, docs: [Document {'id': 'user:3', ...
|
96 | 91 | # STEP_END
|
97 |
| - |
98 | 92 | # Tests for 'query1' step.
|
99 | 93 | # REMOVE_START
|
100 | 94 | assert str(findPaulResult) == (
|
|
104 | 98 | )
|
105 | 99 | # REMOVE_END
|
106 | 100 |
|
107 |
| - |
108 | 101 | # STEP_START query2
|
109 | 102 | citiesResult = r.ft("idx:users").search(
|
110 | 103 | Query("Paul").return_field("$.city", as_field="city")
|
|
113 | 106 | print(citiesResult)
|
114 | 107 | # >>> [Document {'id': 'user:1', 'payload': None, ...
|
115 | 108 | # STEP_END
|
116 |
| - |
117 | 109 | # Tests for 'query2' step.
|
118 | 110 | # REMOVE_START
|
119 | 111 | citiesResult.sort(key=lambda doc: doc['id'])
|
|
124 | 116 | )
|
125 | 117 | # REMOVE_END
|
126 | 118 |
|
127 |
| - |
128 | 119 | # STEP_START query3
|
129 | 120 | req = aggregations.AggregateRequest("*").group_by(
|
130 | 121 | '@city', reducers.count().alias('count')
|
|
134 | 125 | print(aggResult)
|
135 | 126 | # >>> [['city', 'London', 'count', '1'], ['city', 'Tel Aviv', 'count', '2']]
|
136 | 127 | # STEP_END
|
137 |
| - |
138 | 128 | # Tests for 'query3' step.
|
139 | 129 | # REMOVE_START
|
140 | 130 | aggResult.sort(key=lambda row: row[1])
|
|
0 commit comments