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
|
from parameterized import parameterized
from clickhouse_driver import errors
from clickhouse_driver.columns.service import get_column_by_spec
from clickhouse_driver.context import Context
from tests.numpy.testcase import NumpyBaseTestCase
class OtherColumnsTestCase(NumpyBaseTestCase):
def get_column(self, spec):
ctx = Context()
ctx.client_settings = {'strings_as_bytes': False, 'use_numpy': True}
return get_column_by_spec(spec, {'context': ctx})
@parameterized.expand([
("Enum8('hello' = 1, 'world' = 2)", ),
('Decimal(8, 4)', ),
('Array(String)', ),
('Tuple(String)', ),
('SimpleAggregateFunction(any, Int32)', ),
('Map(String, String)', ),
('Array(LowCardinality(String))', )
])
def test_generic_type(self, spec):
col = self.get_column(spec)
self.assertIsNotNone(col)
def test_get_unknown_column(self):
with self.assertRaises(errors.UnknownTypeError) as e:
self.get_column('Unicorn')
self.assertIn('Unicorn', str(e.exception))
|
