pt
/
superset
mirror of https://github.com/apache/superset.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge 31f5340412 into f5843fe588

This commit is contained in:
Balthazar Rouberol 2024-05-05 02:17:01 -03:00 committed by GitHub
parent f5843fe588 31f5340412
commit cd5b256e47
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
superset/db_engine_specs/presto.py
View File

@ -507,7 +507,10 @@ class PrestoBaseEngineSpec(BaseEngineSpec, metaclass=ABCMeta):
}
for col_name, value in zip(col_names, values):
col_type = column_type_by_name.get(col_name)
col_type = None
if col_type_name := column_type_by_name.get(col_name):
if col_type_class := getattr(types, col_type_name, None):
col_type = col_type_class()
if isinstance(col_type, types.DATE):
col_type = Date()

8
tests/unit_tests/db_engine_specs/test_presto.py
View File

@ -116,10 +116,10 @@ def test_get_schema_from_engine_params() -> None:
@pytest.mark.parametrize(
["column_type", "column_value", "expected_value"],
[
(types.DATE(), "2023-05-01", "DATE '2023-05-01'"),
(types.TIMESTAMP(), "2023-05-01", "TIMESTAMP '2023-05-01'"),
(types.VARCHAR(), "2023-05-01", "'2023-05-01'"),
(types.INT(), 1234, "1234"),
("DATE", "2023-05-01", "DATE '2023-05-01'"),
("TIMESTAMP", "2023-05-01", "TIMESTAMP '2023-05-01'"),
("VARCHAR", "2023-05-01", "'2023-05-01'"),
("INT", 1234, "1234"),
],
)
def test_where_latest_partition(