Update SQL engine snapshots

metricflow/dataflow/builder/dataflow_plan_builder.py

@@ -178,8 +178,6 @@ def build_plan(
         optimizers: Sequence[DataflowPlanOptimizer] = (),
     ) -> DataflowPlan:
         """Generate a plan for reading the results of a query with the given spec into a dataframe or table."""
-        assert not query_spec.min_max_only, "`min_max_only` is not supported for metric queries."
-
         metrics_output_node = self._build_metrics_output_node(
             metric_specs=query_spec.metric_specs,
             queried_linkable_specs=query_spec.linkable_specs,

metricflow/test/snapshots/test_query_rendering.py/SqlQueryPlan/BigQuery/test_min_max_only_categorical__plan0.sql

@@ -0,0 +1,73 @@
+-- Calculate min and max
+SELECT
+  MIN(listing__country_latest) AS min
+  , MAX(listing__country_latest) AS max
+FROM (
+  -- Pass Only Elements:
+  --   ['listing__country_latest']
+  SELECT
+    subq_0.listing__country_latest
+  FROM (
+    -- Read Elements From Semantic Model 'listings_latest'
+    SELECT
+      1 AS listings
+      , listings_latest_src_10004.capacity AS largest_listing
+      , listings_latest_src_10004.capacity AS smallest_listing
+      , DATE_TRUNC(listings_latest_src_10004.created_at, day) AS ds__day
+      , DATE_TRUNC(listings_latest_src_10004.created_at, isoweek) AS ds__week
+      , DATE_TRUNC(listings_latest_src_10004.created_at, month) AS ds__month
+      , DATE_TRUNC(listings_latest_src_10004.created_at, quarter) AS ds__quarter
+      , DATE_TRUNC(listings_latest_src_10004.created_at, year) AS ds__year
+      , EXTRACT(year FROM listings_latest_src_10004.created_at) AS ds__extract_year
+      , EXTRACT(quarter FROM listings_latest_src_10004.created_at) AS ds__extract_quarter
+      , EXTRACT(month FROM listings_latest_src_10004.created_at) AS ds__extract_month
+      , EXTRACT(day FROM listings_latest_src_10004.created_at) AS ds__extract_day
+      , IF(EXTRACT(dayofweek FROM listings_latest_src_10004.created_at) = 1, 7, EXTRACT(dayofweek FROM listings_latest_src_10004.created_at) - 1) AS ds__extract_dow
+      , EXTRACT(dayofyear FROM listings_latest_src_10004.created_at) AS ds__extract_doy
+      , DATE_TRUNC(listings_latest_src_10004.created_at, day) AS created_at__day
+      , DATE_TRUNC(listings_latest_src_10004.created_at, isoweek) AS created_at__week
+      , DATE_TRUNC(listings_latest_src_10004.created_at, month) AS created_at__month
+      , DATE_TRUNC(listings_latest_src_10004.created_at, quarter) AS created_at__quarter
+      , DATE_TRUNC(listings_latest_src_10004.created_at, year) AS created_at__year
+      , EXTRACT(year FROM listings_latest_src_10004.created_at) AS created_at__extract_year
+      , EXTRACT(quarter FROM listings_latest_src_10004.created_at) AS created_at__extract_quarter
+      , EXTRACT(month FROM listings_latest_src_10004.created_at) AS created_at__extract_month
+      , EXTRACT(day FROM listings_latest_src_10004.created_at) AS created_at__extract_day
+      , IF(EXTRACT(dayofweek FROM listings_latest_src_10004.created_at) = 1, 7, EXTRACT(dayofweek FROM listings_latest_src_10004.created_at) - 1) AS created_at__extract_dow
+      , EXTRACT(dayofyear FROM listings_latest_src_10004.created_at) AS created_at__extract_doy
+      , listings_latest_src_10004.country AS country_latest
+      , listings_latest_src_10004.is_lux AS is_lux_latest
+      , listings_latest_src_10004.capacity AS capacity_latest
+      , DATE_TRUNC(listings_latest_src_10004.created_at, day) AS listing__ds__day
+      , DATE_TRUNC(listings_latest_src_10004.created_at, isoweek) AS listing__ds__week
+      , DATE_TRUNC(listings_latest_src_10004.created_at, month) AS listing__ds__month
+      , DATE_TRUNC(listings_latest_src_10004.created_at, quarter) AS listing__ds__quarter
+      , DATE_TRUNC(listings_latest_src_10004.created_at, year) AS listing__ds__year
+      , EXTRACT(year FROM listings_latest_src_10004.created_at) AS listing__ds__extract_year
+      , EXTRACT(quarter FROM listings_latest_src_10004.created_at) AS listing__ds__extract_quarter
+      , EXTRACT(month FROM listings_latest_src_10004.created_at) AS listing__ds__extract_month
+      , EXTRACT(day FROM listings_latest_src_10004.created_at) AS listing__ds__extract_day
+      , IF(EXTRACT(dayofweek FROM listings_latest_src_10004.created_at) = 1, 7, EXTRACT(dayofweek FROM listings_latest_src_10004.created_at) - 1) AS listing__ds__extract_dow
+      , EXTRACT(dayofyear FROM listings_latest_src_10004.created_at) AS listing__ds__extract_doy
+      , DATE_TRUNC(listings_latest_src_10004.created_at, day) AS listing__created_at__day
+      , DATE_TRUNC(listings_latest_src_10004.created_at, isoweek) AS listing__created_at__week
+      , DATE_TRUNC(listings_latest_src_10004.created_at, month) AS listing__created_at__month
+      , DATE_TRUNC(listings_latest_src_10004.created_at, quarter) AS listing__created_at__quarter
+      , DATE_TRUNC(listings_latest_src_10004.created_at, year) AS listing__created_at__year
+      , EXTRACT(year FROM listings_latest_src_10004.created_at) AS listing__created_at__extract_year
+      , EXTRACT(quarter FROM listings_latest_src_10004.created_at) AS listing__created_at__extract_quarter
+      , EXTRACT(month FROM listings_latest_src_10004.created_at) AS listing__created_at__extract_month
+      , EXTRACT(day FROM listings_latest_src_10004.created_at) AS listing__created_at__extract_day
+      , IF(EXTRACT(dayofweek FROM listings_latest_src_10004.created_at) = 1, 7, EXTRACT(dayofweek FROM listings_latest_src_10004.created_at) - 1) AS listing__created_at__extract_dow
+      , EXTRACT(dayofyear FROM listings_latest_src_10004.created_at) AS listing__created_at__extract_doy
+      , listings_latest_src_10004.country AS listing__country_latest
+      , listings_latest_src_10004.is_lux AS listing__is_lux_latest
+      , listings_latest_src_10004.capacity AS listing__capacity_latest
+      , listings_latest_src_10004.listing_id AS listing
+      , listings_latest_src_10004.user_id AS user
+      , listings_latest_src_10004.user_id AS listing__user
+    FROM ***************************.dim_listings_latest listings_latest_src_10004
+  ) subq_0
+  GROUP BY
+    listing__country_latest
+) subq_1

metricflow/test/snapshots/test_query_rendering.py/SqlQueryPlan/BigQuery/test_min_max_only_categorical__plan0_optimized.sql

@@ -0,0 +1,14 @@
+-- Calculate min and max
+SELECT
+  MIN(listing__country_latest) AS min
+  , MAX(listing__country_latest) AS max
+FROM (
+  -- Read Elements From Semantic Model 'listings_latest'
+  -- Pass Only Elements:
+  --   ['listing__country_latest']
+  SELECT
+    country AS listing__country_latest
+  FROM ***************************.dim_listings_latest listings_latest_src_10004
+  GROUP BY
+    listing__country_latest
+) subq_3

metricflow/test/snapshots/test_query_rendering.py/SqlQueryPlan/BigQuery/test_min_max_only_time__plan0.sql

@@ -0,0 +1,105 @@
+-- Calculate min and max
+SELECT
+  MIN(booking__paid_at__day) AS min
+  , MAX(booking__paid_at__day) AS max
+FROM (
+  -- Pass Only Elements:
+  --   ['booking__paid_at__day']
+  SELECT
+    subq_0.booking__paid_at__day
+  FROM (
+    -- Read Elements From Semantic Model 'bookings_source'
+    SELECT
+      1 AS bookings
+      , CASE WHEN is_instant THEN 1 ELSE 0 END AS instant_bookings
+      , bookings_source_src_10001.booking_value
+      , bookings_source_src_10001.booking_value AS max_booking_value
+      , bookings_source_src_10001.booking_value AS min_booking_value
+      , bookings_source_src_10001.guest_id AS bookers
+      , bookings_source_src_10001.booking_value AS average_booking_value
+      , bookings_source_src_10001.booking_value AS booking_payments
+      , CASE WHEN referrer_id IS NOT NULL THEN 1 ELSE 0 END AS referred_bookings
+      , bookings_source_src_10001.booking_value AS median_booking_value
+      , bookings_source_src_10001.booking_value AS booking_value_p99
+      , bookings_source_src_10001.booking_value AS discrete_booking_value_p99
+      , bookings_source_src_10001.booking_value AS approximate_continuous_booking_value_p99
+      , bookings_source_src_10001.booking_value AS approximate_discrete_booking_value_p99
+      , bookings_source_src_10001.is_instant
+      , DATE_TRUNC(bookings_source_src_10001.ds, day) AS ds__day
+      , DATE_TRUNC(bookings_source_src_10001.ds, isoweek) AS ds__week
+      , DATE_TRUNC(bookings_source_src_10001.ds, month) AS ds__month
+      , DATE_TRUNC(bookings_source_src_10001.ds, quarter) AS ds__quarter
+      , DATE_TRUNC(bookings_source_src_10001.ds, year) AS ds__year
+      , EXTRACT(year FROM bookings_source_src_10001.ds) AS ds__extract_year
+      , EXTRACT(quarter FROM bookings_source_src_10001.ds) AS ds__extract_quarter
+      , EXTRACT(month FROM bookings_source_src_10001.ds) AS ds__extract_month
+      , EXTRACT(day FROM bookings_source_src_10001.ds) AS ds__extract_day
+      , IF(EXTRACT(dayofweek FROM bookings_source_src_10001.ds) = 1, 7, EXTRACT(dayofweek FROM bookings_source_src_10001.ds) - 1) AS ds__extract_dow
+      , EXTRACT(dayofyear FROM bookings_source_src_10001.ds) AS ds__extract_doy
+      , DATE_TRUNC(bookings_source_src_10001.ds_partitioned, day) AS ds_partitioned__day
+      , DATE_TRUNC(bookings_source_src_10001.ds_partitioned, isoweek) AS ds_partitioned__week
+      , DATE_TRUNC(bookings_source_src_10001.ds_partitioned, month) AS ds_partitioned__month
+      , DATE_TRUNC(bookings_source_src_10001.ds_partitioned, quarter) AS ds_partitioned__quarter
+      , DATE_TRUNC(bookings_source_src_10001.ds_partitioned, year) AS ds_partitioned__year
+      , EXTRACT(year FROM bookings_source_src_10001.ds_partitioned) AS ds_partitioned__extract_year
+      , EXTRACT(quarter FROM bookings_source_src_10001.ds_partitioned) AS ds_partitioned__extract_quarter
+      , EXTRACT(month FROM bookings_source_src_10001.ds_partitioned) AS ds_partitioned__extract_month
+      , EXTRACT(day FROM bookings_source_src_10001.ds_partitioned) AS ds_partitioned__extract_day
+      , IF(EXTRACT(dayofweek FROM bookings_source_src_10001.ds_partitioned) = 1, 7, EXTRACT(dayofweek FROM bookings_source_src_10001.ds_partitioned) - 1) AS ds_partitioned__extract_dow
+      , EXTRACT(dayofyear FROM bookings_source_src_10001.ds_partitioned) AS ds_partitioned__extract_doy
+      , DATE_TRUNC(bookings_source_src_10001.paid_at, day) AS paid_at__day
+      , DATE_TRUNC(bookings_source_src_10001.paid_at, isoweek) AS paid_at__week
+      , DATE_TRUNC(bookings_source_src_10001.paid_at, month) AS paid_at__month
+      , DATE_TRUNC(bookings_source_src_10001.paid_at, quarter) AS paid_at__quarter
+      , DATE_TRUNC(bookings_source_src_10001.paid_at, year) AS paid_at__year
+      , EXTRACT(year FROM bookings_source_src_10001.paid_at) AS paid_at__extract_year
+      , EXTRACT(quarter FROM bookings_source_src_10001.paid_at) AS paid_at__extract_quarter
+      , EXTRACT(month FROM bookings_source_src_10001.paid_at) AS paid_at__extract_month
+      , EXTRACT(day FROM bookings_source_src_10001.paid_at) AS paid_at__extract_day
+      , IF(EXTRACT(dayofweek FROM bookings_source_src_10001.paid_at) = 1, 7, EXTRACT(dayofweek FROM bookings_source_src_10001.paid_at) - 1) AS paid_at__extract_dow
+      , EXTRACT(dayofyear FROM bookings_source_src_10001.paid_at) AS paid_at__extract_doy
+      , bookings_source_src_10001.is_instant AS booking__is_instant
+      , DATE_TRUNC(bookings_source_src_10001.ds, day) AS booking__ds__day
+      , DATE_TRUNC(bookings_source_src_10001.ds, isoweek) AS booking__ds__week
+      , DATE_TRUNC(bookings_source_src_10001.ds, month) AS booking__ds__month
+      , DATE_TRUNC(bookings_source_src_10001.ds, quarter) AS booking__ds__quarter
+      , DATE_TRUNC(bookings_source_src_10001.ds, year) AS booking__ds__year
+      , EXTRACT(year FROM bookings_source_src_10001.ds) AS booking__ds__extract_year
+      , EXTRACT(quarter FROM bookings_source_src_10001.ds) AS booking__ds__extract_quarter
+      , EXTRACT(month FROM bookings_source_src_10001.ds) AS booking__ds__extract_month
+      , EXTRACT(day FROM bookings_source_src_10001.ds) AS booking__ds__extract_day
+      , IF(EXTRACT(dayofweek FROM bookings_source_src_10001.ds) = 1, 7, EXTRACT(dayofweek FROM bookings_source_src_10001.ds) - 1) AS booking__ds__extract_dow
+      , EXTRACT(dayofyear FROM bookings_source_src_10001.ds) AS booking__ds__extract_doy
+      , DATE_TRUNC(bookings_source_src_10001.ds_partitioned, day) AS booking__ds_partitioned__day
+      , DATE_TRUNC(bookings_source_src_10001.ds_partitioned, isoweek) AS booking__ds_partitioned__week
+      , DATE_TRUNC(bookings_source_src_10001.ds_partitioned, month) AS booking__ds_partitioned__month
+      , DATE_TRUNC(bookings_source_src_10001.ds_partitioned, quarter) AS booking__ds_partitioned__quarter
+      , DATE_TRUNC(bookings_source_src_10001.ds_partitioned, year) AS booking__ds_partitioned__year
+      , EXTRACT(year FROM bookings_source_src_10001.ds_partitioned) AS booking__ds_partitioned__extract_year
+      , EXTRACT(quarter FROM bookings_source_src_10001.ds_partitioned) AS booking__ds_partitioned__extract_quarter
+      , EXTRACT(month FROM bookings_source_src_10001.ds_partitioned) AS booking__ds_partitioned__extract_month
+      , EXTRACT(day FROM bookings_source_src_10001.ds_partitioned) AS booking__ds_partitioned__extract_day
+      , IF(EXTRACT(dayofweek FROM bookings_source_src_10001.ds_partitioned) = 1, 7, EXTRACT(dayofweek FROM bookings_source_src_10001.ds_partitioned) - 1) AS booking__ds_partitioned__extract_dow
+      , EXTRACT(dayofyear FROM bookings_source_src_10001.ds_partitioned) AS booking__ds_partitioned__extract_doy
+      , DATE_TRUNC(bookings_source_src_10001.paid_at, day) AS booking__paid_at__day
+      , DATE_TRUNC(bookings_source_src_10001.paid_at, isoweek) AS booking__paid_at__week
+      , DATE_TRUNC(bookings_source_src_10001.paid_at, month) AS booking__paid_at__month
+      , DATE_TRUNC(bookings_source_src_10001.paid_at, quarter) AS booking__paid_at__quarter
+      , DATE_TRUNC(bookings_source_src_10001.paid_at, year) AS booking__paid_at__year
+      , EXTRACT(year FROM bookings_source_src_10001.paid_at) AS booking__paid_at__extract_year
+      , EXTRACT(quarter FROM bookings_source_src_10001.paid_at) AS booking__paid_at__extract_quarter
+      , EXTRACT(month FROM bookings_source_src_10001.paid_at) AS booking__paid_at__extract_month
+      , EXTRACT(day FROM bookings_source_src_10001.paid_at) AS booking__paid_at__extract_day
+      , IF(EXTRACT(dayofweek FROM bookings_source_src_10001.paid_at) = 1, 7, EXTRACT(dayofweek FROM bookings_source_src_10001.paid_at) - 1) AS booking__paid_at__extract_dow
+      , EXTRACT(dayofyear FROM bookings_source_src_10001.paid_at) AS booking__paid_at__extract_doy
+      , bookings_source_src_10001.listing_id AS listing
+      , bookings_source_src_10001.guest_id AS guest
+      , bookings_source_src_10001.host_id AS host
+      , bookings_source_src_10001.listing_id AS booking__listing
+      , bookings_source_src_10001.guest_id AS booking__guest
+      , bookings_source_src_10001.host_id AS booking__host
+    FROM ***************************.fct_bookings bookings_source_src_10001
+  ) subq_0
+  GROUP BY
+    booking__paid_at__day
+) subq_1

metricflow/test/snapshots/test_query_rendering.py/SqlQueryPlan/BigQuery/test_min_max_only_time__plan0_optimized.sql

@@ -0,0 +1,14 @@
+-- Calculate min and max
+SELECT
+  MIN(booking__paid_at__day) AS min
+  , MAX(booking__paid_at__day) AS max
+FROM (
+  -- Read Elements From Semantic Model 'bookings_source'
+  -- Pass Only Elements:
+  --   ['booking__paid_at__day']
+  SELECT
+    DATE_TRUNC(paid_at, day) AS booking__paid_at__day
+  FROM ***************************.fct_bookings bookings_source_src_10001
+  GROUP BY
+    booking__paid_at__day
+) subq_3