Fixed broken test

apache · Oct 19, 2022 · 40a5454 · 40a5454
1 parent 2f8f77c
commit 40a5454
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/tests/integration_tests/db_engine_specs/hive_tests.py b/tests/integration_tests/db_engine_specs/hive_tests.py
@@ -270,7 +270,6 @@ def test_df_to_sql_schema_definition(mock_upload_to_s3, mock_g):
         "datetime": pd.Series(dtype="datetime64[ns]")
     })
     expected_schema_definition = "`bool` BOOLEAN, `float64` DOUBLE, `int64` BIGINT, `object` STRING, `datetime` TIMESTAMP"
-
     with app.app_context():
         HiveEngineSpec.df_to_sql(
             mock_database,
@@ -279,13 +278,8 @@ def test_df_to_sql_schema_definition(mock_upload_to_s3, mock_g):
             {"if_exists": "replace"},
         )
 
-    mock_execute.assert_any_call(
-        f"""
-        CREATE TABLE {table_name} ({expected_schema_definition})
-        STORED AS PARQUET
-        LOCATION :{mock_location}
-        """
-    )
+    _, call_args, _ = mock_execute.mock_calls[1]
+    assert expected_schema_definition in str(call_args[0])
     app.config = config