mirror of
https://github.com/hanzo-ds/driver.git
synced 2026-10-30 20:32:40 +00:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
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))
|