mirror of
https://github.com/hanzo-ds/connect.git
synced 2026-10-30 20:31:24 +00:00
zeekay
999d38466d
Imports read `datastore_connect`, distribution `datastore-connect`, so nothing has to import another product's name to talk to Hanzo Datastore. Cut from upstream 0.8.17 — the version this stack already ran — rather than from upstream HEAD, so it is a PURE RENAME with no behaviour change. Forking from HEAD would have dragged versions along with the name (connect alone jumped 0.8.17 -> 1.6.0, a major), which is a different and much riskier change wearing a rename's clothes. X-ClickHouse-* headers are untouched: they are wire protocol the server sends and expects, and a different token shape from the module name.
35 lines
1.6 KiB
Python
35 lines
1.6 KiB
Python
from typing import Callable
|
|
|
|
from datastore_connect.driver import Client
|
|
|
|
|
|
def test_point_column(test_client: Client, table_context: Callable):
|
|
with table_context('point_column_test', ['key Int32', 'point Point']):
|
|
data = [[1, (3.55, 3.55)], [2, (4.55, 4.55)]]
|
|
test_client.insert('point_column_test', data)
|
|
|
|
query_result = test_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(test_client: Client, 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)]]]
|
|
test_client.insert('ring_column_test', data)
|
|
|
|
query_result = test_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(test_client: Client, table_context: Callable):
|
|
with table_context('polygon_column_test', ['key Int32', 'polygon Polygon']):
|
|
res = test_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]
|
|
test_client.insert('polygon_column_test', [(1, pg), (4, pg)])
|
|
query_result = test_client.query('SELECT key, polygon FROM polygon_column_test WHERE key = 4')
|
|
assert query_result.first_row[1] == pg
|