diff --git a/tests/providers/openlineage/test_sqlparser.py b/tests/providers/openlineage/test_sqlparser.py index 22bd0674229ec..30e74eefcb556 100644 --- a/tests/providers/openlineage/test_sqlparser.py +++ b/tests/providers/openlineage/test_sqlparser.py @@ -283,12 +283,9 @@ def test_generate_openlineage_metadata_from_sql(self, mock_parse, parser_returns ] sql = """INSERT INTO popular_orders_day_of_week (order_day_of_week) - SELECT EXTRACT(ISODOW FROM order_placed_on) AS order_day_of_week - FROM top_delivery_times - --irrelevant comment - ) - ; - """ + SELECT EXTRACT(ISODOW FROM order_placed_on) AS order_day_of_week + FROM top_delivery_times --irrelevant comment + );""" hook.get_conn.return_value.cursor.return_value.fetchall.side_effect = returned_rows @@ -314,10 +311,9 @@ def test_generate_openlineage_metadata_from_sql(self, mock_parse, parser_returns mock_parse.return_value = mock_sql_meta formatted_sql = """INSERT INTO popular_orders_day_of_week (order_day_of_week) - SELECT EXTRACT(ISODOW FROM order_placed_on) AS order_day_of_week - FROM top_delivery_times - -)""" + SELECT EXTRACT(ISODOW FROM order_placed_on) AS order_day_of_week + FROM top_delivery_times + )""" expected_schema = "PUBLIC" if parser_returns_schema else "ANOTHER_SCHEMA" expected = OperatorLineage( inputs=[