diff --git a/tests/unit_tests/datasets/dao/dao_tests.py b/tests/unit_tests/datasets/dao/dao_tests.py index 10115ba9374d5..350425d08e897 100644 --- a/tests/unit_tests/datasets/dao/dao_tests.py +++ b/tests/unit_tests/datasets/dao/dao_tests.py @@ -72,6 +72,7 @@ def test_datasource_find_by_id_skip_base_filter_not_found( ) assert result is None + def test_datasource_find_by_ids_skip_base_filter(session_with_data: Session) -> None: from superset.connectors.sqla.models import SqlaTable from superset.datasets.dao import DatasetDAO @@ -87,7 +88,10 @@ def test_datasource_find_by_ids_skip_base_filter(session_with_data: Session) -> assert ["my_sqla_table"] == list(map(lambda x: x.table_name, result)) assert isinstance(result[0], SqlaTable) -def test_datasource_find_by_ids_skip_base_filter_not_found(session_with_data: Session) -> None: + +def test_datasource_find_by_ids_skip_base_filter_not_found( + session_with_data: Session, +) -> None: from superset.datasets.dao import DatasetDAO result = DatasetDAO.find_by_ids(