@@ -98,93 +98,93 @@ def _gen_example_objects(schema_name: str) -> ExampleObjects:
9898 seed_model_name = f"{ schema_name } .seed_model"
9999
100100 full_model_def = f"""MODEL (
101- name { full_model_name } ,
102- kind FULL,
103- cron '@daily',
104- grain item_id,
105- audits (assert_positive_order_ids),
106- );
107-
108- SELECT
109- item_id,
110- COUNT(DISTINCT id) AS num_orders,
111- FROM
112- { incremental_model_name }
113- GROUP BY item_id
101+ name { full_model_name } ,
102+ kind FULL,
103+ cron '@daily',
104+ grain item_id,
105+ audits (assert_positive_order_ids),
106+ );
107+
108+ SELECT
109+ item_id,
110+ COUNT(DISTINCT id) AS num_orders,
111+ FROM
112+ { incremental_model_name }
113+ GROUP BY item_id
114114 """
115115
116116 incremental_model_def = f"""MODEL (
117- name { incremental_model_name } ,
118- kind INCREMENTAL_BY_TIME_RANGE (
119- time_column event_date
120- ),
121- start '2020-01-01',
122- cron '@daily',
123- grain (id, event_date)
124- );
125-
126- SELECT
127- id,
128- item_id,
129- event_date,
130- FROM
131- { seed_model_name }
132- WHERE
133- event_date BETWEEN @start_date AND @end_date
117+ name { incremental_model_name } ,
118+ kind INCREMENTAL_BY_TIME_RANGE (
119+ time_column event_date
120+ ),
121+ start '2020-01-01',
122+ cron '@daily',
123+ grain (id, event_date)
124+ );
125+
126+ SELECT
127+ id,
128+ item_id,
129+ event_date,
130+ FROM
131+ { seed_model_name }
132+ WHERE
133+ event_date BETWEEN @start_date AND @end_date
134134 """
135135
136136 seed_model_def = f"""MODEL (
137- name { seed_model_name } ,
138- kind SEED (
139- path '../seeds/seed_data.csv'
140- ),
141- columns (
142- id INTEGER,
143- item_id INTEGER,
144- event_date DATE
145- ),
146- grain (id, event_date)
147- );
137+ name { seed_model_name } ,
138+ kind SEED (
139+ path '../seeds/seed_data.csv'
140+ ),
141+ columns (
142+ id INTEGER,
143+ item_id INTEGER,
144+ event_date DATE
145+ ),
146+ grain (id, event_date)
147+ );
148148 """
149149
150150 audit_def = """AUDIT (
151- name assert_positive_order_ids,
152- );
151+ name assert_positive_order_ids,
152+ );
153153
154- SELECT *
155- FROM @this_model
156- WHERE
157- item_id < 0
154+ SELECT *
155+ FROM @this_model
156+ WHERE
157+ item_id < 0
158158 """
159159
160160 seed_data = """id,item_id,event_date
161- 1,2,2020-01-01
162- 2,1,2020-01-01
163- 3,3,2020-01-03
164- 4,1,2020-01-04
165- 5,1,2020-01-05
166- 6,1,2020-01-06
167- 7,1,2020-01-07
168- """
161+ 1,2,2020-01-01
162+ 2,1,2020-01-01
163+ 3,3,2020-01-03
164+ 4,1,2020-01-04
165+ 5,1,2020-01-05
166+ 6,1,2020-01-06
167+ 7,1,2020-01-07
168+ """
169169
170170 test_def = f"""test_example_full_model:
171- model: { full_model_name }
172- inputs:
173- { incremental_model_name } :
174- rows:
175- - id: 1
176- item_id: 1
177- - id: 2
178- item_id: 1
179- - id: 3
180- item_id: 2
181- outputs:
182- query:
183- rows:
184- - item_id: 1
185- num_orders: 2
186- - item_id: 2
187- num_orders: 1
171+ model: { full_model_name }
172+ inputs:
173+ { incremental_model_name } :
174+ rows:
175+ - id: 1
176+ item_id: 1
177+ - id: 2
178+ item_id: 1
179+ - id: 3
180+ item_id: 2
181+ outputs:
182+ query:
183+ rows:
184+ - item_id: 1
185+ num_orders: 2
186+ - item_id: 2
187+ num_orders: 1
188188 """
189189
190190 return ExampleObjects (
0 commit comments