TPC-DS-OBT Q13 — obt_distinct_items

TPC-DS-OBT benchmark · all TPC-DS-OBT queries

Count of distinct items sold

Representative SQL

Rendered for DataFusion with default parameters.

SELECT
  AVG(quantity),
  AVG(ext_sales_price),
  AVG(ext_wholesale_cost),
  SUM(ext_wholesale_cost)
FROM tpcds_sales_returns_obt AS obt
WHERE
  (
    store_s_store_sk = store_sk
    AND sold_date_sk = sold_date_d_date_sk
    AND sold_date_d_year = 2001
    AND (
      (
        ship_hdemo_sk = bill_hdemo_hd_demo_sk
        AND bill_cdemo_cd_demo_sk = ship_cdemo_sk
        AND bill_cdemo_cd_marital_status = '1'
        AND bill_cdemo_cd_education_status = '1'
        AND sales_price BETWEEN 100.00 AND 150.00
        AND bill_hdemo_hd_dep_count = 3
      )
      OR (
        ship_hdemo_sk = bill_hdemo_hd_demo_sk
        AND bill_cdemo_cd_demo_sk = ship_cdemo_sk
        AND bill_cdemo_cd_marital_status = '1'
        AND bill_cdemo_cd_education_status = '1'
        AND sales_price BETWEEN 50.00 AND 100.00
        AND bill_hdemo_hd_dep_count = 1
      )
      OR (
        ship_hdemo_sk = bill_hdemo_hd_demo_sk
        AND bill_cdemo_cd_demo_sk = ship_cdemo_sk
        AND bill_cdemo_cd_marital_status = '1'
        AND bill_cdemo_cd_education_status = '1'
        AND sales_price BETWEEN 150.00 AND 200.00
        AND bill_hdemo_hd_dep_count = 1
      )
    )
    AND (
      (
        ship_addr_sk = bill_addr_ca_address_sk
        AND bill_addr_ca_country = 'United States'
        AND bill_addr_ca_state IN ('3', '3', '3')
        AND net_profit BETWEEN 100 AND 200
      )
      OR (
        ship_addr_sk = bill_addr_ca_address_sk
        AND bill_addr_ca_country = 'United States'
        AND bill_addr_ca_state IN ('3', '3', '3')
        AND net_profit BETWEEN 150 AND 300
      )
      OR (
        ship_addr_sk = bill_addr_ca_address_sk
        AND bill_addr_ca_country = 'United States'
        AND bill_addr_ca_state IN ('3', '3', '3')
        AND net_profit BETWEEN 50 AND 250
      )
    )
  )
  AND channel = 'store'

Representative DataFrame

Expression family (Polars, PySpark, DataFusion), default parameters.

def impl(ctx: DataFrameContext) -> Any:
        table = ctx.get_table(OBT_TABLE)
        for column, op, value in spec.filters:
            table = table.filter(_FILTER_OPS[op](ctx.col(column), ctx.lit(value)))
        aggregates = [getattr(ctx.col(column), func)().alias(alias) for alias, column, func in spec.aggregations]
        if spec.group_keys:
            result = table.group_by(*spec.group_keys).agg(*aggregates)
        else:
            result = table.select(*aggregates)
        if spec.sort_keys:
            descending = list(spec.descending) if spec.descending else [False] * len(spec.sort_keys)
            result = result.sort(list(spec.sort_keys), descending=descending)
        if spec.limit:
            result = result.limit(spec.limit)
        return result

Get the exact query for your run

The blocks above are representative. To get the precise statement a specific platform runs at a specific scale, substitute your own values:

from benchbox import TPCDSOBT

bench = TPCDSOBT(scale_factor=1.0)            # your scale factor
queries = bench.get_queries(dialect="datafusion")   # your platform's dialect; drop dialect= for native SQL
print(queries['13'])

For the DataFrame form:

from benchbox.core.query_catalog import get_dataframe_render

print(get_dataframe_render('tpcds_obt', '13', family="expression").source)  # or family="pandas"

Source

Query text: benchbox/core/tpcds_obt/queries.py