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
35 lines
1.7 KiB
Python
35 lines
1.7 KiB
Python
from collections.abc import Callable
|
|
|
|
from clickhouse_connect.driver import Client
|
|
|
|
|
|
def test_point_column(param_client: Client, call, table_context: Callable):
|
|
with table_context("point_column_test", ["key Int32", "point Point"]):
|
|
data = [[1, (3.55, 3.55)], [2, (4.55, 4.55)]]
|
|
call(param_client.insert, "point_column_test", data)
|
|
|
|
query_result = call(param_client.query, "SELECT * FROM point_column_test ORDER BY key").result_rows
|
|
assert len(query_result) == 2
|
|
assert query_result[0] == (1, (3.55, 3.55))
|
|
assert query_result[1] == (2, (4.55, 4.55))
|
|
|
|
|
|
def test_ring_column(param_client: Client, call, table_context: Callable):
|
|
with table_context("ring_column_test", ["key Int32", "ring Ring"]):
|
|
data = [[1, [(5.522, 58.472), (3.55, 3.55)]], [2, [(4.55, 4.55)]]]
|
|
call(param_client.insert, "ring_column_test", data)
|
|
|
|
query_result = call(param_client.query, "SELECT * FROM ring_column_test ORDER BY key").result_rows
|
|
assert len(query_result) == 2
|
|
assert query_result[0] == (1, [(5.522, 58.472), (3.55, 3.55)])
|
|
assert query_result[1] == (2, [(4.55, 4.55)])
|
|
|
|
|
|
def test_polygon_column(param_client: Client, call, table_context: Callable):
|
|
with table_context("polygon_column_test", ["key Int32", "polygon Polygon"]):
|
|
res = call(param_client.query, "SELECT readWKTPolygon('POLYGON ((-64.8 32.3, -65.5 18.3, -80.3 25.2, -64.8 32.3))') as polygon")
|
|
pg = res.first_row[0]
|
|
call(param_client.insert, "polygon_column_test", [(1, pg), (4, pg)])
|
|
query_result = call(param_client.query, "SELECT key, polygon FROM polygon_column_test WHERE key = 4")
|
|
assert query_result.first_row[1] == pg
|