From fa5c083ed59389a04a9542187941586d2cd881d5 Mon Sep 17 00:00:00 2001 From: justinpark Date: Fri, 20 Dec 2024 10:29:34 -0800 Subject: [PATCH] pylint --- tests/unit_tests/databases/utils_test.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) 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())])"