2020-06-06 20:56:32 +03:00
|
|
|
import unittest
|
|
|
|
|
|
|
|
|
|
from infi.clickhouse_orm import *
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class IndexesTest(unittest.TestCase):
|
|
|
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
|
self.database = Database('test-db', log_statements=True)
|
2020-06-06 21:06:29 +03:00
|
|
|
if self.database.server_version < (20, 1, 2, 4):
|
2020-06-06 20:56:32 +03:00
|
|
|
raise unittest.SkipTest('ClickHouse version too old')
|
|
|
|
|
|
|
|
|
|
def tearDown(self):
|
|
|
|
|
self.database.drop_database()
|
|
|
|
|
|
|
|
|
|
def test_all_index_types(self):
|
|
|
|
|
self.database.create_table(ModelWithIndexes)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class ModelWithIndexes(Model):
|
|
|
|
|
|
|
|
|
|
date = DateField()
|
|
|
|
|
f1 = Int32Field()
|
|
|
|
|
f2 = StringField()
|
|
|
|
|
|
|
|
|
|
i1 = Index(f1, type=Index.minmax(), granularity=1)
|
|
|
|
|
i2 = Index(f1, type=Index.set(1000), granularity=2)
|
|
|
|
|
i3 = Index(f2, type=Index.ngrambf_v1(3, 256, 2, 0), granularity=1)
|
|
|
|
|
i4 = Index(F.lower(f2), type=Index.tokenbf_v1(256, 2, 0), granularity=2)
|
|
|
|
|
i5 = Index((F.toQuarter(date), f2), type=Index.bloom_filter(), granularity=3)
|
|
|
|
|
|
|
|
|
|
engine = MergeTree('date', ('date',))
|
