Skip to content

Commit

Permalink
Update snapshots for other SQL engines
Browse files Browse the repository at this point in the history
  • Loading branch information
@courtneyholcomb
courtneyholcomb committed Jan 30, 2025
1 parent 2a98d84 commit 5266f30
Show file tree
Hide file tree
Showing 36 changed files with 5,004 additions and 384 deletions.
34 changes: 12 additions & 22 deletions ...napshots/test_custom_granularity.py/SqlPlan/BigQuery/test_custom_offset_window__plan0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -137,28 +137,18 @@ FROM (
WITH cte_2 AS (
-- Get Custom Granularity Bounds
SELECT
time_spine_src_28006.ds AS ds__day
, DATETIME_TRUNC(time_spine_src_28006.ds, isoweek) AS ds__week
, DATETIME_TRUNC(time_spine_src_28006.ds, month) AS ds__month
, DATETIME_TRUNC(time_spine_src_28006.ds, quarter) AS ds__quarter
, DATETIME_TRUNC(time_spine_src_28006.ds, year) AS ds__year
, EXTRACT(year FROM time_spine_src_28006.ds) AS ds__extract_year
, EXTRACT(quarter FROM time_spine_src_28006.ds) AS ds__extract_quarter
, EXTRACT(month FROM time_spine_src_28006.ds) AS ds__extract_month
, EXTRACT(day FROM time_spine_src_28006.ds) AS ds__extract_day
, IF(EXTRACT(dayofweek FROM time_spine_src_28006.ds) = 1, 7, EXTRACT(dayofweek FROM time_spine_src_28006.ds) - 1) AS ds__extract_dow
, EXTRACT(dayofyear FROM time_spine_src_28006.ds) AS ds__extract_doy
, time_spine_src_28006.martian_day AS ds__martian_day
subq_6.ds__day
, subq_6.ds__martian_day
, FIRST_VALUE(subq_6.ds__day) OVER (
PARTITION BY subq_6.ds__martian_day
ORDER BY subq_6.ds__day
ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING
) AS ds__martian_day__first_value
) AS ds__day__first_value
, LAST_VALUE(subq_6.ds__day) OVER (
PARTITION BY subq_6.ds__martian_day
ORDER BY subq_6.ds__day
ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING
) AS ds__martian_day__last_value
) AS ds__day__last_value
, ROW_NUMBER() OVER (
PARTITION BY subq_6.ds__martian_day
ORDER BY subq_6.ds__day
Expand All @@ -185,28 +175,28 @@ FROM (
SELECT
cte_2.ds__day AS ds__day
, CASE
WHEN DATE_ADD(CAST(subq_8.ds__martian_day__first_value__lead AS DATETIME), INTERVAL cte_2.ds__day__row_number - 1 day) <= subq_8.ds__martian_day__last_value__lead
THEN DATE_ADD(CAST(subq_8.ds__martian_day__first_value__lead AS DATETIME), INTERVAL cte_2.ds__day__row_number - 1 day)
WHEN DATE_ADD(CAST(subq_8.ds__day__first_value__lead AS DATETIME), INTERVAL cte_2.ds__day__row_number - 1 day) <= subq_8.ds__day__last_value__lead
THEN DATE_ADD(CAST(subq_8.ds__day__first_value__lead AS DATETIME), INTERVAL cte_2.ds__day__row_number - 1 day)
ELSE NULL
END AS ds__day__lead
FROM cte_2 cte_2
INNER JOIN (
-- Offset Custom Granularity Bounds
SELECT
subq_7.ds__martian_day
, LEAD(subq_7.ds__martian_day__first_value, 1) OVER (ORDER BY subq_7.ds__martian_day) AS ds__martian_day__first_value__lead
, LEAD(subq_7.ds__martian_day__last_value, 1) OVER (ORDER BY subq_7.ds__martian_day) AS ds__martian_day__last_value__lead
, LEAD(subq_7.ds__day__first_value, 1) OVER (ORDER BY subq_7.ds__martian_day) AS ds__day__first_value__lead
, LEAD(subq_7.ds__day__last_value, 1) OVER (ORDER BY subq_7.ds__martian_day) AS ds__day__last_value__lead
FROM (
-- Get Unique Rows for Custom Granularity Bounds
SELECT
cte_2.ds__martian_day
, cte_2.ds__martian_day__first_value
, cte_2.ds__martian_day__last_value
, cte_2.ds__day__first_value
, cte_2.ds__day__last_value
FROM cte_2 cte_2
GROUP BY
ds__martian_day
, ds__martian_day__first_value
, ds__martian_day__last_value
, ds__day__first_value
, ds__day__last_value
) subq_7
) subq_8
ON
Expand Down
20 changes: 10 additions & 10 deletions ...est_custom_granularity.py/SqlPlan/BigQuery/test_custom_offset_window__plan0_optimized.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -26,12 +26,12 @@ FROM (
PARTITION BY martian_day
ORDER BY ds
ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING
) AS ds__martian_day__first_value
) AS ds__day__first_value
, LAST_VALUE(ds) OVER (
PARTITION BY martian_day
ORDER BY ds
ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING
) AS ds__martian_day__last_value
) AS ds__day__last_value
, ROW_NUMBER() OVER (
PARTITION BY martian_day
ORDER BY ds
Expand All @@ -42,28 +42,28 @@ FROM (
SELECT
cte_6.ds__day AS ds__day
, CASE
WHEN DATE_ADD(CAST(subq_24.ds__martian_day__first_value__lead AS DATETIME), INTERVAL cte_6.ds__day__row_number - 1 day) <= subq_24.ds__martian_day__last_value__lead
THEN DATE_ADD(CAST(subq_24.ds__martian_day__first_value__lead AS DATETIME), INTERVAL cte_6.ds__day__row_number - 1 day)
WHEN DATE_ADD(CAST(subq_24.ds__day__first_value__lead AS DATETIME), INTERVAL cte_6.ds__day__row_number - 1 day) <= subq_24.ds__day__last_value__lead
THEN DATE_ADD(CAST(subq_24.ds__day__first_value__lead AS DATETIME), INTERVAL cte_6.ds__day__row_number - 1 day)
ELSE NULL
END AS ds__day__lead
FROM cte_6 cte_6
INNER JOIN (
-- Offset Custom Granularity Bounds
SELECT
ds__martian_day
, LEAD(ds__martian_day__first_value, 1) OVER (ORDER BY ds__martian_day) AS ds__martian_day__first_value__lead
, LEAD(ds__martian_day__last_value, 1) OVER (ORDER BY ds__martian_day) AS ds__martian_day__last_value__lead
, LEAD(ds__day__first_value, 1) OVER (ORDER BY ds__martian_day) AS ds__day__first_value__lead
, LEAD(ds__day__last_value, 1) OVER (ORDER BY ds__martian_day) AS ds__day__last_value__lead
FROM (
-- Get Unique Rows for Custom Granularity Bounds
SELECT
ds__martian_day
, ds__martian_day__first_value
, ds__martian_day__last_value
, ds__day__first_value
, ds__day__last_value
FROM cte_6 cte_6
GROUP BY
ds__martian_day
, ds__martian_day__first_value
, ds__martian_day__last_value
, ds__day__first_value
, ds__day__last_value
) subq_23
) subq_24
ON
Expand Down
34 changes: 12 additions & 22 deletions ...y.py/SqlPlan/BigQuery/test_custom_offset_window_with_granularity_and_date_part__plan0.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -151,28 +151,18 @@ FROM (
WITH cte_2 AS (
-- Get Custom Granularity Bounds
SELECT
time_spine_src_28006.ds AS ds__day
, DATETIME_TRUNC(time_spine_src_28006.ds, isoweek) AS ds__week
, DATETIME_TRUNC(time_spine_src_28006.ds, month) AS ds__month
, DATETIME_TRUNC(time_spine_src_28006.ds, quarter) AS ds__quarter
, DATETIME_TRUNC(time_spine_src_28006.ds, year) AS ds__year
, EXTRACT(year FROM time_spine_src_28006.ds) AS ds__extract_year
, EXTRACT(quarter FROM time_spine_src_28006.ds) AS ds__extract_quarter
, EXTRACT(month FROM time_spine_src_28006.ds) AS ds__extract_month
, EXTRACT(day FROM time_spine_src_28006.ds) AS ds__extract_day
, IF(EXTRACT(dayofweek FROM time_spine_src_28006.ds) = 1, 7, EXTRACT(dayofweek FROM time_spine_src_28006.ds) - 1) AS ds__extract_dow
, EXTRACT(dayofyear FROM time_spine_src_28006.ds) AS ds__extract_doy
, time_spine_src_28006.martian_day AS ds__martian_day
subq_6.ds__day
, subq_6.ds__martian_day
, FIRST_VALUE(subq_6.ds__day) OVER (
PARTITION BY subq_6.ds__martian_day
ORDER BY subq_6.ds__day
ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING
) AS ds__martian_day__first_value
) AS ds__day__first_value
, LAST_VALUE(subq_6.ds__day) OVER (
PARTITION BY subq_6.ds__martian_day
ORDER BY subq_6.ds__day
ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING
) AS ds__martian_day__last_value
) AS ds__day__last_value
, ROW_NUMBER() OVER (
PARTITION BY subq_6.ds__martian_day
ORDER BY subq_6.ds__day
Expand All @@ -199,28 +189,28 @@ FROM (
SELECT
cte_2.ds__day AS ds__day
, CASE
WHEN DATE_ADD(CAST(subq_8.ds__martian_day__first_value__lead AS DATETIME), INTERVAL cte_2.ds__day__row_number - 1 day) <= subq_8.ds__martian_day__last_value__lead
THEN DATE_ADD(CAST(subq_8.ds__martian_day__first_value__lead AS DATETIME), INTERVAL cte_2.ds__day__row_number - 1 day)
WHEN DATE_ADD(CAST(subq_8.ds__day__first_value__lead AS DATETIME), INTERVAL cte_2.ds__day__row_number - 1 day) <= subq_8.ds__day__last_value__lead
THEN DATE_ADD(CAST(subq_8.ds__day__first_value__lead AS DATETIME), INTERVAL cte_2.ds__day__row_number - 1 day)
ELSE NULL
END AS ds__day__lead
FROM cte_2 cte_2
INNER JOIN (
-- Offset Custom Granularity Bounds
SELECT
subq_7.ds__martian_day
, LEAD(subq_7.ds__martian_day__first_value, 1) OVER (ORDER BY subq_7.ds__martian_day) AS ds__martian_day__first_value__lead
, LEAD(subq_7.ds__martian_day__last_value, 1) OVER (ORDER BY subq_7.ds__martian_day) AS ds__martian_day__last_value__lead
, LEAD(subq_7.ds__day__first_value, 1) OVER (ORDER BY subq_7.ds__martian_day) AS ds__day__first_value__lead
, LEAD(subq_7.ds__day__last_value, 1) OVER (ORDER BY subq_7.ds__martian_day) AS ds__day__last_value__lead
FROM (
-- Get Unique Rows for Custom Granularity Bounds
SELECT
cte_2.ds__martian_day
, cte_2.ds__martian_day__first_value
, cte_2.ds__martian_day__last_value
, cte_2.ds__day__first_value
, cte_2.ds__day__last_value
FROM cte_2 cte_2
GROUP BY
ds__martian_day
, ds__martian_day__first_value
, ds__martian_day__last_value
, ds__day__first_value
, ds__day__last_value
) subq_7
) subq_8
ON
Expand Down
20 changes: 10 additions & 10 deletions ...an/BigQuery/test_custom_offset_window_with_granularity_and_date_part__plan0_optimized.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,12 +31,12 @@ FROM (
PARTITION BY martian_day
ORDER BY ds
ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING
) AS ds__martian_day__first_value
) AS ds__day__first_value
, LAST_VALUE(ds) OVER (
PARTITION BY martian_day
ORDER BY ds
ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING
) AS ds__martian_day__last_value
) AS ds__day__last_value
, ROW_NUMBER() OVER (
PARTITION BY martian_day
ORDER BY ds
Expand All @@ -47,28 +47,28 @@ FROM (
SELECT
cte_6.ds__day AS ds__day
, CASE
WHEN DATE_ADD(CAST(subq_25.ds__martian_day__first_value__lead AS DATETIME), INTERVAL cte_6.ds__day__row_number - 1 day) <= subq_25.ds__martian_day__last_value__lead
THEN DATE_ADD(CAST(subq_25.ds__martian_day__first_value__lead AS DATETIME), INTERVAL cte_6.ds__day__row_number - 1 day)
WHEN DATE_ADD(CAST(subq_25.ds__day__first_value__lead AS DATETIME), INTERVAL cte_6.ds__day__row_number - 1 day) <= subq_25.ds__day__last_value__lead
THEN DATE_ADD(CAST(subq_25.ds__day__first_value__lead AS DATETIME), INTERVAL cte_6.ds__day__row_number - 1 day)
ELSE NULL
END AS ds__day__lead
FROM cte_6 cte_6
INNER JOIN (
-- Offset Custom Granularity Bounds
SELECT
ds__martian_day
, LEAD(ds__martian_day__first_value, 1) OVER (ORDER BY ds__martian_day) AS ds__martian_day__first_value__lead
, LEAD(ds__martian_day__last_value, 1) OVER (ORDER BY ds__martian_day) AS ds__martian_day__last_value__lead
, LEAD(ds__day__first_value, 1) OVER (ORDER BY ds__martian_day) AS ds__day__first_value__lead
, LEAD(ds__day__last_value, 1) OVER (ORDER BY ds__martian_day) AS ds__day__last_value__lead
FROM (
-- Get Unique Rows for Custom Granularity Bounds
SELECT
ds__martian_day
, ds__martian_day__first_value
, ds__martian_day__last_value
, ds__day__first_value
, ds__day__last_value
FROM cte_6 cte_6
GROUP BY
ds__martian_day
, ds__martian_day__first_value
, ds__martian_day__last_value
, ds__day__first_value
, ds__day__last_value
) subq_24
) subq_25
ON
Expand Down
Loading

0 comments on commit 5266f30

Please sign in to comment.