python/adbc_driver_postgresql/benchmarks/benchmarks.py [96:125]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        FROM GENERATE_SERIES(1, {row_count}) temp(generated)""",
        # TODO: does an index matter, do we want to force PostgreSQL
        # to update statistics?
    ]

    param_data = {
        "row_count": [10_000, 100_000, 1_000_000],
        "data_type": ["INT", "BIGINT", "FLOAT", "DOUBLE PRECISION"],
    }

    param_names = list(param_data.keys())
    params = list(param_data.values())

    def setup_cache(self) -> None:
        self.uri = os.environ["ADBC_POSTGRESQL_TEST_URI"]
        with psycopg.connect(self.uri) as conn:
            with conn.cursor() as cursor:
                for row_count, data_type in itertools.product(*self.params):
                    table_name = self._make_table_name(row_count, data_type)
                    for query in self.SETUP_QUERIES:
                        cursor.execute(
                            query.format(
                                table_name=table_name,
                                row_count=row_count,
                                data_type=data_type,
                            )
                        )

    def _make_table_name(self, row_count: int, data_type: str) -> str:
        return (f"bench_{row_count}_{data_type.replace(' ', '_')}").lower()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



python/adbc_driver_postgresql/benchmarks/benchmarks.py [148:175]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        """,
    ]

    param_data = {
        "row_count": [10_000, 100_000, 1_000_000],
        "data_type": ["INT", "BIGINT", "FLOAT", "DOUBLE PRECISION"],
    }

    param_names = list(param_data.keys())
    params = list(param_data.values())

    def setup_cache(self) -> None:
        self.uri = os.environ["ADBC_POSTGRESQL_TEST_URI"]
        with psycopg.connect(self.uri) as conn:
            with conn.cursor() as cursor:
                for row_count, data_type in itertools.product(*self.params):
                    table_name = self._make_table_name(row_count, data_type)
                    for query in self.SETUP_QUERIES:
                        cursor.execute(
                            query.format(
                                table_name=table_name,
                                row_count=row_count,
                                data_type=data_type,
                            )
                        )

    def _make_table_name(self, row_count: int, data_type: str) -> str:
        return (f"bench_{row_count}_{data_type.replace(' ', '_')}").lower()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -