forked from feldera/feldera
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_empty_set.py
More file actions
265 lines (205 loc) · 8.66 KB
/
test_empty_set.py
File metadata and controls
265 lines (205 loc) · 8.66 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
from tests.aggregate_tests.aggtst_base import TstView
# Array
class aggtst_count_emp_test(TstView):
def __init__(self):
self.data = [{"count": 0}]
self.sql = """CREATE MATERIALIZED VIEW count_emp AS SELECT
COUNT (*) AS count
FROM array_tbl
WHERE FALSE"""
class aggtst_count_col_emp_test(TstView):
def __init__(self):
self.data = [{"count_col": 0}]
self.sql = """CREATE MATERIALIZED VIEW count_col_emp AS SELECT
COUNT(c1) AS count_col
FROM array_tbl
WHERE FALSE"""
class aggtst_max_emp_test(TstView):
def __init__(self):
self.data = [{"max": None}]
self.sql = """CREATE MATERIALIZED VIEW max_emp AS SELECT
MAX(c1) FILTER(WHERE c1 IS NULL) AS max
FROM array_tbl"""
class aggtst_min_emp_test(TstView):
def __init__(self):
self.data = [{"min": None}]
self.sql = """CREATE MATERIALIZED VIEW min_emp AS SELECT
MIN(c1) FILTER(WHERE c1 IS NULL) AS min
FROM array_tbl"""
class aggtst_arg_max_emp_test(TstView):
def __init__(self):
self.data = [{"arg_max": None}]
self.sql = """CREATE MATERIALIZED VIEW arg_max_emp AS SELECT
ARG_MAX(c1, c2) FILTER(WHERE c1 IS NULL) AS arg_max
FROM array_tbl"""
class aggtst_arg_min_emp_test(TstView):
def __init__(self):
self.data = [{"arg_min": None}]
self.sql = """CREATE MATERIALIZED VIEW arg_min_emp AS SELECT
ARG_MIN(c1, c2) FILTER(WHERE c1 IS NULL) AS arg_min
FROM array_tbl"""
class aggtst_some_emp_test(TstView):
def __init__(self):
self.data = [{"sme": None}]
self.sql = """CREATE MATERIALIZED VIEW some_emp AS SELECT
SOME(c1 > c2) AS sme
FROM array_tbl
WHERE FALSE"""
class aggtst_every_emp_test(TstView):
def __init__(self):
self.data = [{"evry": None}]
self.sql = """CREATE MATERIALIZED VIEW every_emp AS SELECT
EVERY(c1 > c2) AS evry
FROM array_tbl
WHERE FALSE"""
class aggtst_arr_agg_emp_test(TstView):
def __init__(self):
self.data = [{"arr_agg": []}]
self.sql = """CREATE MATERIALIZED VIEW arr_agg_emp AS SELECT
ARRAY_AGG(c1) AS arr_agg
FROM array_tbl
WHERE FALSE"""
# ROW
class aggtst_count_col_emp_test_row(TstView):
def __init__(self):
self.data = [{"count_col": 0}]
self.sql = """CREATE MATERIALIZED VIEW count_col_emp_row AS SELECT
COUNT(DISTINCT ROW(c1, c2, c3)) AS count_col
FROM row_tbl
WHERE FALSE"""
class aggtst_max_emp_test_row(TstView):
def __init__(self):
self.data = [{"max": None}]
self.sql = """CREATE MATERIALIZED VIEW max_emp_row AS SELECT
MAX(ROW(c1, c2, c3)) FILTER(WHERE c1 IS NULL) AS max
FROM row_tbl"""
class aggtst_min_emp_test_row(TstView):
def __init__(self):
self.data = [{"min": None}]
self.sql = """CREATE MATERIALIZED VIEW min_emp_row AS SELECT
MIN(ROW(c1, c2, c3)) AS min
FROM row_tbl
WHERE FALSE"""
class aggtst_argmin_emp_test_row(TstView):
def __init__(self):
self.data = [{"argmin": None}]
self.sql = """CREATE MATERIALIZED VIEW argmin_emp_test_row AS SELECT
ARG_MIN(ROW(c1, c2, c3), c2) AS argmin
FROM row_tbl
WHERE FALSE"""
class aggtst_argmax_emp_test_row(TstView):
def __init__(self):
self.data = [{"argmax": None}]
self.sql = """CREATE MATERIALIZED VIEW argmax_emp_test_row AS SELECT
ARG_MAX(ROW(c1, c2, c3), c2) AS argmax
FROM row_tbl
WHERE FALSE"""
class aggtst_arrgg_emp_test_row(TstView):
def __init__(self):
self.data = [{"arragg": []}]
self.sql = """CREATE MATERIALIZED VIEW arrgg_emp_test_row AS SELECT
ARRAY_AGG(ROW(c2, c3)) AS arragg
FROM row_tbl
WHERE FALSE"""
class aggtst_some_emp_test_row(TstView):
def __init__(self):
self.data = [{"sme": None}]
self.sql = """CREATE MATERIALIZED VIEW some_emp_row AS SELECT
SOME(c1 > SAFE_CAST(c2 AS INT)) AS sme
FROM row_tbl"""
class aggtst_every_emp_test_row(TstView):
def __init__(self):
self.data = [{"evry": None}]
self.sql = """CREATE MATERIALIZED VIEW every_emp_row AS SELECT
EVERY(c1 > SAFE_CAST(c2 AS INT)) AS evry
FROM row_tbl"""
# Map
class aggtst_count_emp_test_map(TstView):
def __init__(self):
self.data = [{"count": 0}]
self.sql = """CREATE MATERIALIZED VIEW count_emp_map AS SELECT
COUNT (*) AS count
FROM map_tbl
WHERE FALSE"""
class aggtst_count_col_emp_test_map(TstView):
def __init__(self):
self.data = [{"count_col": 0}]
self.sql = """CREATE MATERIALIZED VIEW count_col_emp_map AS SELECT
COUNT(c1) AS count_col
FROM map_tbl
WHERE c1 IS NULL"""
class aggtst_max_emp_test_map(TstView):
def __init__(self):
self.data = [{"max": None}]
self.sql = """CREATE MATERIALIZED VIEW max_emp_map AS SELECT
MAX(c1) FILTER(WHERE c1 IS NULL) AS max
FROM map_tbl"""
class aggtst_min_emp_test_map(TstView):
def __init__(self):
self.data = [{"min": None}]
self.sql = """CREATE MATERIALIZED VIEW min_emp_map AS SELECT
MIN(c1) AS min
FROM map_tbl
WHERE c1 = c2"""
class aggtst_argmin_emp_test_map(TstView):
def __init__(self):
self.data = [{"argmin": None}]
self.sql = """CREATE MATERIALIZED VIEW argmin_emp_test_map AS SELECT
ARG_MIN(c1, c2) AS argmin
FROM map_tbl
WHERE FALSE"""
class aggtst_argmax_emp_test_map(TstView):
def __init__(self):
self.data = [{"argmax": None}]
self.sql = """CREATE MATERIALIZED VIEW argmax_emp_test_map AS SELECT
ARG_MAX(c2, c1) AS argmax
FROM map_tbl
WHERE FALSE"""
class aggtst_arrgg_emp_test_map(TstView):
def __init__(self):
self.data = [{"arragg": []}]
self.sql = """CREATE MATERIALIZED VIEW arrgg_emp_test_map AS SELECT
ARRAY_AGG(c1) AS arragg
FROM map_tbl
WHERE FALSE"""
class aggtst_some_emp_test_map(TstView):
def __init__(self):
self.data = [{"sme": None}]
self.sql = """CREATE MATERIALIZED VIEW some_emp_map AS SELECT
SOME(c1 > c2) AS sme
FROM map_tbl
WHERE FALSE"""
class aggtst_every_emp_test_map(TstView):
def __init__(self):
self.data = [{"evry": None}]
self.sql = """CREATE MATERIALIZED VIEW every_emp_map AS SELECT
EVERY(c1 > c2) AS evry
FROM map_tbl
WHERE FALSE"""
# Int
class aggtst_avg_emp_test(TstView):
def __init__(self):
self.data = [{"c1_avg": None, "c2_avg": None}]
self.sql = """CREATE MATERIALIZED VIEW avg_emp AS SELECT
AVG(c1) FILTER(WHERE c1 < 1) AS c1_avg,
AVG(c2) FILTER(WHERE c2 < 2) AS c2_avg
FROM int_tbl"""
class aggtst_sum_emp_test(TstView):
def __init__(self):
self.data = [{"sum": None}]
self.sql = """CREATE MATERIALIZED VIEW sum_emp AS SELECT
SUM(c1) FILTER(WHERE c1 < 1) AS sum
FROM int_tbl"""
class aggtst_stdev_samp_emp_test(TstView):
def __init__(self):
self.data = [{"stdev_samp": None}]
self.sql = """CREATE MATERIALIZED VIEW stdev_samp_emp AS SELECT
STDDEV_SAMP(c1) FILTER(WHERE c1 < 1) AS stdev_samp
FROM int_tbl"""
class aggtst_stdev_pop_emp_test(TstView):
def __init__(self):
self.data = [{"stdev_pop": None}]
self.sql = """CREATE MATERIALIZED VIEW stdev_pop_emp AS SELECT
STDDEV_POP(c1) AS stdev_pop
FROM int_tbl
WHERE c1 = 7"""