Method test_union

test/sql/test_statement_params.py:133–183

(self, connection)

            eq_(cs(stmt3.params({"x": 9})).all(), [9])

    def test_union(self, connection):
        a = self.tables.a

        cs = connection.scalars
        for _ in range(3):
            x1 = random.randint(1, 10)
            x2 = random.randint(1, 10)

            eq_(
                cs(
                    select(a)
                    .where(a.c.data == x1)
                    .union_all(select(a).where(a.c.data == x2))
                    .order_by(a.c.data)
                ).all(),
                sorted([x1, x2]),
            )

            x1 = random.randint(1, 10)
            x2 = random.randint(1, 10)
            stmt = (
                select(a, literal(1).label("ord"))
                .where(a.c.data == bindparam("x", x1))
                .union_all(
                    select(a, literal(2)).where(a.c.data == bindparam("y", x2))
                )
                .order_by("ord")
            )
            eq_(cs(stmt).all(), [x1, x2])

            x1a = random.randint(1, 10)
            eq_(cs(stmt.params({"x": x1a})).all(), [x1a, x2])

            x2 = random.randint(1, 10)
            eq_(cs(stmt, {"y": x2}).all(), [x1, x2])

            x1 = random.randint(1, 10)
            x2 = random.randint(1, 10)
            eq_(cs(stmt.params({"x": x1}), {"y": x2}).all(), [x1, x2])

            x1 = random.randint(1, 10)
            x2 = random.randint(1, 10)
            stmt2 = (
                stmt.params(x=x1)
                .subquery()
                .select()
                .params(y=x2)
                .order_by("ord")
            )
            eq_(cs(stmt2).all(), [x1, x2])
            eq_(cs(stmt2.params({"x": x1}).params({"y": x2})).all(), [x1, x2])

    def test_text(self, connection):
        a = self.tables.a