|
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