summaryrefslogtreecommitdiffstats
path: root/tests/dataframe/integration
diff options
context:
space:
mode:
Diffstat (limited to 'tests/dataframe/integration')
-rw-r--r--tests/dataframe/integration/dataframe_validator.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/dataframe/integration/dataframe_validator.py b/tests/dataframe/integration/dataframe_validator.py
index c84a342..22d4982 100644
--- a/tests/dataframe/integration/dataframe_validator.py
+++ b/tests/dataframe/integration/dataframe_validator.py
@@ -135,9 +135,9 @@ class DataFrameValidator(unittest.TestCase):
data=district_data, schema=cls.sqlglot_district_schema
)
cls.df_district.createOrReplaceTempView("district")
- sqlglot.schema.add_table("employee", cls.sqlglot_employee_schema)
- sqlglot.schema.add_table("store", cls.sqlglot_store_schema)
- sqlglot.schema.add_table("district", cls.sqlglot_district_schema)
+ sqlglot.schema.add_table("employee", cls.sqlglot_employee_schema, dialect="spark")
+ sqlglot.schema.add_table("store", cls.sqlglot_store_schema, dialect="spark")
+ sqlglot.schema.add_table("district", cls.sqlglot_district_schema, dialect="spark")
def setUp(self) -> None:
warnings.filterwarnings("ignore", category=ResourceWarning)