mirror of
https://github.com/hanzo-ds/connect.git
synced 2026-10-30 20:31:24 +00:00
Joe SpadolaandGitHub
55dcd7db4c
* migrate from pylint to ruff * ruff format/lint switch over * add formatting commit * update changelog
32 lines
859 B
Python
32 lines
859 B
Python
import datetime
|
|
|
|
from clickhouse_connect.datatypes.registry import get_from_name
|
|
from clickhouse_connect.driver.insert import InsertContext
|
|
from clickhouse_connect.tools.datagen import fixed_len_ascii_str
|
|
|
|
|
|
def test_block_size():
|
|
data = [(1, (datetime.date(2020, 5, 2), datetime.datetime(2020, 5, 2, 10, 5, 2)))]
|
|
ctx = InsertContext(
|
|
"fake_table",
|
|
["key", "date_tuple"],
|
|
[
|
|
get_from_name("UInt64"),
|
|
get_from_name("Tuple(Date, DateTime)"),
|
|
],
|
|
data,
|
|
)
|
|
assert ctx.block_row_count == 262144
|
|
|
|
data = [(x, fixed_len_ascii_str(400)) for x in range(5000)]
|
|
ctx = InsertContext(
|
|
"fake_table",
|
|
["key", "big_str"],
|
|
[
|
|
get_from_name("Int32"),
|
|
get_from_name("String"),
|
|
],
|
|
data,
|
|
)
|
|
assert ctx.block_row_count == 8192
|