diff --git a/python-sdk/tests_integration/sql/operators/test_load_file.py b/python-sdk/tests_integration/sql/operators/test_load_file.py index 8ffaf6d15..5552dc2b6 100644 --- a/python-sdk/tests_integration/sql/operators/test_load_file.py +++ b/python-sdk/tests_integration/sql/operators/test_load_file.py @@ -834,7 +834,7 @@ def test_load_file_bigquery_error_out(sample_dag, database_table_fixture): indirect=True, ids=["snowflake", "bigquery", "postgresql", "sqlite", "redshift", "duckdb", "mysql"], ) -@pytest.mark.parametrize("file_type", ["parquet", "ndjson", "json", "csv", "xlsx"]) +@pytest.mark.parametrize("file_type", ["parquet", "ndjson", "json", "csv"]) def test_load_file(sample_dag, database_table_fixture, file_type): db, test_table = database_table_fixture @@ -875,7 +875,7 @@ def test_load_file(sample_dag, database_table_fixture, file_type): indirect=True, ids=["mssql"], ) -@pytest.mark.parametrize("file_type", ["parquet", "ndjson", "json", "csv"]) +@pytest.mark.parametrize("file_type", ["parquet", "ndjson", "json", "csv", "xlsx"]) def test_load_file_for_mssql(sample_dag, database_table_fixture, file_type): """Test loading a file into MSSQL table and then exporting the table to dataframe works successfully""" db, test_table = database_table_fixture