class TestTable(self.base): date = Column(types.Date, primary_key=True) x = Column(types.Int32) y = Column(types.Int32) __table_args__ = (engines.AggregatingMergeTree(partition_by=date, order_by=(date, x)), )
def test_aggregating_merge_tree(self): engine = engines.AggregatingMergeTree(partition_by='x', order_by='x', primary_key='x') with self._test_table(engine) as (table, engine): self.assertIsInstance(engine, engines.AggregatingMergeTree) self.assertEqual(engine.partition_by.columns, [table.c.x]) self.assertEqual(engine.order_by.columns, [table.c.x]) self.assertEqual(engine.primary_key.columns, [table.c.x])