mirror of
https://github.com/hanzo-ds/driver.git
synced 2026-10-30 20:32:40 +00:00
zeekay
eb74c996a6
Imports read `datastore_driver`, distribution `datastore-driver`, so nothing has to import another product's name to talk to Hanzo Datastore. Cut from upstream 0.2.9 — 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.
84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
import math
|
|
|
|
from tests.testcase import BaseTestCase
|
|
from datastore_driver import errors
|
|
|
|
|
|
class FloatTestCase(BaseTestCase):
|
|
def test_chop_to_type(self):
|
|
with self.create_table('a Float32, b Float64'):
|
|
data = [
|
|
(3.4028235e38, 3.4028235e38),
|
|
(3.4028235e39, 3.4028235e39),
|
|
(-3.4028235e39, 3.4028235e39),
|
|
(1, 2)
|
|
]
|
|
|
|
with self.assertRaises(errors.TypeMismatchError) as e:
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data
|
|
)
|
|
|
|
self.assertIn('Column a', str(e.exception))
|
|
|
|
def test_simple(self):
|
|
with self.create_table('a Float32, b Float64'):
|
|
data = [
|
|
(3.4028235e38, 3.4028235e38),
|
|
(3.4028235e39, 3.4028235e39),
|
|
(-3.4028235e39, 3.4028235e39),
|
|
(1, 2)
|
|
]
|
|
self.client.execute(
|
|
'INSERT INTO test (a, b) VALUES', data, types_check=True
|
|
)
|
|
|
|
query = 'SELECT * FROM test'
|
|
inserted = self.emit_cli(query)
|
|
self.assertEqual(
|
|
inserted, (
|
|
'3.4028235e38\t3.4028235e38\n'
|
|
'inf\t3.4028235e39\n'
|
|
'-inf\t3.4028235e39\n'
|
|
'1\t2\n'
|
|
)
|
|
)
|
|
|
|
inserted = self.client.execute(query)
|
|
self.assertEqual(inserted, [
|
|
(3.4028234663852886e+38, 3.4028235e38),
|
|
(float('inf'), 3.4028235e39),
|
|
(-float('inf'), 3.4028235e39),
|
|
(1, 2)
|
|
])
|
|
|
|
def test_nullable(self):
|
|
with self.create_table('a Nullable(Float32)'):
|
|
data = [(None, ), (0.5, ), (None, ), (1.5, )]
|
|
self.client.execute(
|
|
'INSERT INTO test (a) VALUES', data
|
|
)
|
|
|
|
query = 'SELECT * FROM test'
|
|
inserted = self.emit_cli(query)
|
|
self.assertEqual(inserted, '\\N\n0.5\n\\N\n1.5\n')
|
|
|
|
inserted = self.client.execute(query)
|
|
self.assertEqual(inserted, data)
|
|
|
|
def test_nan(self):
|
|
with self.create_table('a Float32'):
|
|
data = [(float('nan'), ), (0.5, )]
|
|
self.client.execute(
|
|
'INSERT INTO test (a) VALUES', data
|
|
)
|
|
|
|
query = 'SELECT * FROM test'
|
|
inserted = self.emit_cli(query)
|
|
self.assertEqual(inserted, 'nan\n0.5\n')
|
|
|
|
inserted = self.client.execute(query)
|
|
self.assertEqual(len(inserted), 2)
|
|
self.assertTrue(math.isnan(inserted[0][0]))
|
|
self.assertEqual(inserted[1][0], 0.5)
|