diff --git a/tests/unit_tests/databases/utils_test.py b/tests/unit_tests/databases/utils_test.py index 30e833013a907..36dec4af1ee2d 100644 --- a/tests/unit_tests/databases/utils_test.py +++ b/tests/unit_tests/databases/utils_test.py @@ -20,7 +20,8 @@ from sqlalchemy.sql import sqltypes from trino.sqlalchemy import datatype -from superset.databases.utils import make_url_safe, get_col_type +from superset.databases.utils import get_col_type, make_url_safe + def test_make_url_safe_string(session: Session) -> None: """ @@ -40,6 +41,7 @@ def test_make_url_safe_url(session: Session) -> None: uri_safe = make_url_safe(uri) assert uri_safe == uri + def test_get_col_type_primary(session: Session) -> None: """ Test getting a column type @@ -53,6 +55,7 @@ def test_get_col_type_primary(session: Session) -> None: } assert get_col_type(col) == "INTEGER()" + def test_get_col_type_row_and_map(session: Session) -> None: """ Test getting a column type @@ -61,12 +64,12 @@ def test_get_col_type_row_and_map(session: Session) -> None: "name": "id", "type": datatype.ROW( [ - ('name', sqltypes.VARCHAR()), - ('age', sqltypes.INTEGER()), + ("name", sqltypes.VARCHAR()), + ("age", sqltypes.INTEGER()), ] ), "nullable": False, "default": None, "comment": None, } - assert get_col_type(col) == "ROW([('name', VARCHAR()), ('age', INTEGER())])" \ No newline at end of file + assert get_col_type(col) == "ROW([('name', VARCHAR()), ('age', INTEGER())])"