Method test_aggregate_order_by_one

test/sql/test_functions.py:1148–1181

(self, style)

    @testing.variation("style", ["none", "inline", "within_group"])
    def test_aggregate_order_by_one(self, style):
        table = Table(
            "table1", MetaData(), Column("a", Integer), Column("b", Integer)
        )
        expr = func.array_agg(table.c.a).aggregate_order_by(table.c.b.desc())
        stmt = select(expr)

        if style.none:
            dialect = default.DefaultDialect()
            dialect.aggregate_order_by_style = AggregateOrderByStyle.NONE
            with expect_raises_message(
                exc.CompileError,
                "this dialect does not support ORDER BY "
                "within an aggregate function",
            ):
                stmt.compile(dialect=dialect)
        elif style.within_group:
            dialect = default.DefaultDialect()
            dialect.aggregate_order_by_style = (
                AggregateOrderByStyle.WITHIN_GROUP
            )
            self.assert_compile(
                stmt,
                "SELECT array_agg(table1.a) "
                "WITHIN GROUP (ORDER BY table1.b DESC) "
                "AS array_agg_1 FROM table1",
                dialect=dialect,
            )
        else:
            self.assert_compile(
                stmt,
                "SELECT array_agg(table1.a ORDER BY table1.b DESC) "
                "AS array_agg_1 FROM table1",
            )

    @testing.variation("style", ["inline", "within_group"])
    def test_aggregate_order_by_two(self, style):