Skip to content

Commit

Permalink
Fix flake8
Browse files Browse the repository at this point in the history
  • Loading branch information
chethanuk committed Jun 3, 2022
1 parent b361e8a commit 985d8f3
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 additions & 2 deletions tests/system/providers/snowflake/example_snowflake.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,13 +54,13 @@
# [START howto_operator_snowflake]

with DAG(
ENV_ID,
DAG_ID,
start_date=datetime(2021, 1, 1),
default_args={'snowflake_conn_id': SNOWFLAKE_CONN_ID},
tags=['example'],
catchup=False,
) as dag:

# [START snowflake_example_dag]
snowflake_op_sql_str = SnowflakeOperator(
task_id='snowflake_op_sql_str',
sql=CREATE_TABLE_SQL_STRING,
Expand All @@ -80,6 +80,8 @@
role=SNOWFLAKE_ROLE,
)

snowflake_op_sql_list = SnowflakeOperator(task_id='snowflake_op_sql_list', sql=SQL_LIST)

snowflake_op_sql_multiple_stmts = SnowflakeOperator(
task_id='snowflake_op_sql_multiple_stmts',
sql=SQL_MULTIPLE_STMTS,
Expand Down Expand Up @@ -127,3 +129,10 @@
]
>> slack_report
)
# [END snowflake_example_dag]


from tests.system.utils import get_test_run # noqa: E402

# Needed to run the example DAG with pytest (see: tests/system/README.md#run_via_pytest)
test_run = get_test_run(dag)

0 comments on commit 985d8f3

Please sign in to comment.