forked from feldera/feldera
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_array_count_col.py
More file actions
67 lines (55 loc) · 2.59 KB
/
test_array_count_col.py
File metadata and controls
67 lines (55 loc) · 2.59 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
from .aggtst_base import TstView
class aggtst_array_count_col(TstView):
def __init__(self):
# Validated on Postgres
self.data = [{"c1": 4, "c2": 3, "c3": 3}]
self.sql = """CREATE MATERIALIZED VIEW array_count_col AS SELECT
COUNT(c1) AS c1, COUNT(c2) AS c2, COUNT(c3) AS c3
FROM array_tbl"""
class aggtst_array_count_col_gby(TstView):
def __init__(self):
# Validated on Postgres
self.data = [
{"id": 0, "c1": 2, "c2": 1, "c3": 2},
{"id": 1, "c1": 2, "c2": 2, "c3": 1},
]
self.sql = """CREATE MATERIALIZED VIEW array_count_col_gby AS SELECT
id, COUNT(c1) AS c1, COUNT(c2) AS c2, COUNT(c3) AS c3
FROM array_tbl
GROUP BY id"""
class aggtst_array_count_col_distinct(TstView):
def __init__(self):
# Validated on Postgres
self.data = [{"c1": 3, "c2": 3, "c3": 3}]
self.sql = """CREATE MATERIALIZED VIEW array_count_col_distinct AS SELECT
COUNT(DISTINCT c1) AS c1, COUNT(DISTINCT c2) AS c2, COUNT(DISTINCT c3) AS c3
FROM array_tbl"""
class aggtst_array_count_col_distinct_gby(TstView):
def __init__(self):
# Validated on Postgres
self.data = [
{"id": 0, "c1": 2, "c2": 1, "c3": 2},
{"id": 1, "c1": 2, "c2": 2, "c3": 1},
]
self.sql = """CREATE MATERIALIZED VIEW array_count_col_distinct_gby AS SELECT
id, COUNT(DISTINCT c1) AS c1, COUNT(DISTINCT c2) AS c2, COUNT(DISTINCT c3) AS c3
FROM array_tbl
GROUP BY id"""
class aggtst_array_count_col_where(TstView):
def __init__(self):
# Validated on Postgres
self.data = [{"c1": 2, "c2": 2, "c3": 1}]
self.sql = """CREATE MATERIALIZED VIEW array_count_col_where AS SELECT
COUNT(c1) FILTER(WHERE c1 < c2) AS c1, COUNT(c2) FILTER(WHERE c1 < c2) AS c2, COUNT(c3) FILTER(WHERE c1 < c2) AS c3
FROM array_tbl"""
class aggtst_array_count_where_groupby(TstView):
def __init__(self):
# Validated on Postgres
self.data = [
{"id": 0, "c1": 1, "c2": 1, "c3": 1},
{"id": 1, "c1": 1, "c2": 1, "c3": 0},
]
self.sql = """CREATE MATERIALIZED VIEW array_count_col_where_gby AS SELECT
id, COUNT(c1) FILTER(WHERE c1 < c2) AS c1, COUNT(c2) FILTER(WHERE c1 < c2) AS c2, COUNT(c3) FILTER(WHERE c1 < c2) AS c3
FROM array_tbl
GROUP BY id"""