1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
import math
from tests.testcase import BaseTestCase
from clickhouse_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)
|
