Method test_splice_horizontally

test/sql/test_resultset.py:163–250

(self, connection, filters)

    @testing.requires.insert_executemany_returning
    @testing.variation("filters", ["unique", "sliced", "plain"])
    def test_splice_horizontally(self, connection, filters):
        users = self.tables.users
        addresses = self.tables.addresses

        if filters.unique:
            r1 = connection.execute(
                users.insert().returning(users.c.user_name),
                [
                    dict(user_id=1, user_name="john"),
                    dict(user_id=2, user_name="john"),
                ],
            )
            r2 = connection.execute(
                addresses.insert().returning(
                    addresses.c.address,
                ),
                [
                    dict(address_id=1, user_id=1, address="foo@bar.com"),
                    dict(address_id=2, user_id=2, address="foo@bar.com"),
                ],
            )
        else:
            r1 = connection.execute(
                users.insert().returning(users.c.user_name, users.c.user_id),
                [
                    dict(user_id=1, user_name="john"),
                    dict(user_id=2, user_name="jack"),
                ],
            )
            r2 = connection.execute(
                addresses.insert().returning(
                    addresses.c.address_id,
                    addresses.c.address,
                    addresses.c.user_id,
                ),
                [
                    dict(address_id=1, user_id=1, address="foo@bar.com"),
                    dict(address_id=2, user_id=2, address="bar@bat.com"),
                ],
            )

        if filters.sliced:
            r1 = r1.columns(users.c.user_name)
            r2 = r2.columns(addresses.c.address, addresses.c.user_id)
        elif filters.unique:
            r1 = r1.unique()
            r2 = r2.unique()

        rows = r1.splice_horizontally(r2).all()

        if filters.sliced:
            eq_(
                rows,
                [
                    ("john", "foo@bar.com", 1),
                    ("jack", "bar@bat.com", 2),
                ],
            )