mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 10:22:10 +00:00
74 lines
1.7 KiB
Python
74 lines
1.7 KiB
Python
import pytest
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
|
|
node1 = cluster.add_instance(
|
|
"node1",
|
|
main_configs=["configs/clusters.xml"],
|
|
with_zookeeper=True,
|
|
use_old_analyzer=True,
|
|
)
|
|
node2 = cluster.add_instance(
|
|
"node2",
|
|
main_configs=["configs/clusters.xml"],
|
|
with_zookeeper=True,
|
|
use_old_analyzer=True,
|
|
)
|
|
|
|
|
|
def insert_data(node, table_name):
|
|
node.query(
|
|
"""INSERT INTO {}
|
|
VALUES (bitmapBuild(cast([1,2,3,4,5,6,7,8,9,10] as Array(UInt32))));""".format(
|
|
table_name
|
|
)
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def start_cluster():
|
|
try:
|
|
cluster.start()
|
|
yield cluster
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
def test_groupBitmapAnd_on_distributed_table(start_cluster):
|
|
local_table_name = "bitmap_column_expr_test"
|
|
distributed_table_name = "bitmap_column_expr_test_dst"
|
|
cluster_name = "awesome_cluster"
|
|
|
|
for node in (node1, node2):
|
|
node.query(
|
|
"""CREATE TABLE {}
|
|
(
|
|
z AggregateFunction(groupBitmap, UInt32)
|
|
)
|
|
ENGINE = MergeTree()
|
|
ORDER BY tuple()""".format(
|
|
local_table_name
|
|
)
|
|
)
|
|
|
|
node.query(
|
|
"""CREATE TABLE {}
|
|
(
|
|
z AggregateFunction(groupBitmap, UInt32)
|
|
)
|
|
ENGINE = Distributed('{}', 'default', '{}')""".format(
|
|
distributed_table_name, cluster_name, local_table_name
|
|
)
|
|
)
|
|
insert_data(node1, local_table_name)
|
|
|
|
expected = "10"
|
|
|
|
for node in (node1, node2):
|
|
result = node.query(
|
|
"select groupBitmapAnd(z) FROM {};".format(distributed_table_name)
|
|
).strip()
|
|
assert result == expected
|