diff --git a/superset/advanced_data_type/plugins/internet_address.py b/superset/advanced_data_type/plugins/internet_address.py index 3a759a4d35d..9d3eabfd56c 100644 --- a/superset/advanced_data_type/plugins/internet_address.py +++ b/superset/advanced_data_type/plugins/internet_address.py @@ -66,9 +66,9 @@ def cidr_func(req: AdvancedDataTypeRequest) -> AdvancedDataTypeResponse: else: resp["display_value"] = ", ".join( map( # noqa: C417 - lambda x: ( - f"{x['start']} - {x['end']}" if isinstance(x, dict) else str(x) - ), + lambda x: f"{x['start']} - {x['end']}" + if isinstance(x, dict) + else str(x), resp["values"], ) ) diff --git a/superset/advanced_data_type/plugins/internet_port.py b/superset/advanced_data_type/plugins/internet_port.py index bc0fd24a753..25451b8abcb 100644 --- a/superset/advanced_data_type/plugins/internet_port.py +++ b/superset/advanced_data_type/plugins/internet_port.py @@ -95,9 +95,9 @@ def port_translation_func(req: AdvancedDataTypeRequest) -> AdvancedDataTypeRespo else: resp["display_value"] = ", ".join( map( # noqa: C417 - lambda x: ( - f"{x['start']} - {x['end']}" if isinstance(x, dict) else str(x) - ), + lambda x: f"{x['start']} - {x['end']}" + if isinstance(x, dict) + else str(x), resp["values"], ) ) diff --git a/superset/databases/api.py b/superset/databases/api.py index aef93d45a63..abfb99827b2 100644 --- a/superset/databases/api.py +++ b/superset/databases/api.py @@ -650,9 +650,8 @@ class DatabaseRestApi(BaseSupersetModelRestApi): @protect() @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.sync-permissions" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".sync-permissions", log_to_statsd=False, ) def sync_permissions(self, pk: int, **kwargs: Any) -> FlaskResponse: @@ -910,9 +909,8 @@ class DatabaseRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.table_metadata_deprecated" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".table_metadata_deprecated", log_to_statsd=False, ) def table_metadata_deprecated( @@ -975,9 +973,8 @@ class DatabaseRestApi(BaseSupersetModelRestApi): @statsd_metrics @deprecated(deprecated_in="4.0") @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.table_extra_metadata_deprecated" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".table_extra_metadata_deprecated", log_to_statsd=False, ) def table_extra_metadata_deprecated( @@ -1039,9 +1036,8 @@ class DatabaseRestApi(BaseSupersetModelRestApi): @protect() @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.table_metadata" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".table_metadata", log_to_statsd=False, ) def table_metadata(self, pk: int) -> FlaskResponse: @@ -1129,9 +1125,8 @@ class DatabaseRestApi(BaseSupersetModelRestApi): @protect() @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.table_extra_metadata" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".table_extra_metadata", log_to_statsd=False, ) def table_extra_metadata(self, pk: int) -> FlaskResponse: @@ -1273,9 +1268,8 @@ class DatabaseRestApi(BaseSupersetModelRestApi): @protect() @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.test_connection" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".test_connection", log_to_statsd=False, ) @requires_json @@ -1328,9 +1322,8 @@ class DatabaseRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.related_objects" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".related_objects", log_to_statsd=False, ) def related_objects(self, pk: int) -> Response: @@ -1860,9 +1853,8 @@ class DatabaseRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.function_names" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".function_names", log_to_statsd=False, ) def function_names(self, pk: int) -> Response: @@ -2016,9 +2008,8 @@ class DatabaseRestApi(BaseSupersetModelRestApi): @protect() @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.validate_parameters" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".validate_parameters", log_to_statsd=False, ) @requires_json @@ -2074,9 +2065,8 @@ class DatabaseRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.schemas_access_for_file_upload" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".schemas_access_for_file_upload", log_to_statsd=False, ) def schemas_access_for_file_upload(self, pk: int) -> Response: diff --git a/superset/datasets/api.py b/superset/datasets/api.py index 4801e3afcc5..4e542f6973a 100644 --- a/superset/datasets/api.py +++ b/superset/datasets/api.py @@ -713,9 +713,8 @@ class DatasetRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.detect_datetime_formats" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + ".detect_datetime_formats", log_to_statsd=False, ) def detect_datetime_formats(self, pk: int) -> Response: @@ -796,9 +795,8 @@ class DatasetRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.related_objects" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".related_objects", log_to_statsd=False, ) def related_objects(self, id_or_uuid: str) -> Response: @@ -1056,9 +1054,8 @@ class DatasetRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.get_or_create_dataset" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".get_or_create_dataset", log_to_statsd=False, ) def get_or_create_dataset(self) -> Response: @@ -1310,9 +1307,9 @@ class DatasetRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.get_drill_info" - ), + action=lambda self, + *args, + **kwargs: f"{self.__class__.__name__}.get_drill_info", log_to_statsd=False, ) def get_drill_info(self, pk: int, **kwargs: Any) -> Response: diff --git a/superset/datasource/api.py b/superset/datasource/api.py index f04db51e886..cb3bea7f480 100644 --- a/superset/datasource/api.py +++ b/superset/datasource/api.py @@ -59,9 +59,8 @@ class DatasourceRestApi(BaseSupersetApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.get_column_values" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".get_column_values", log_to_statsd=False, ) def get_column_values( @@ -236,9 +235,8 @@ class DatasourceRestApi(BaseSupersetApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.validate_expression" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".validate_expression", log_to_statsd=False, ) def validate_expression( diff --git a/superset/queries/api.py b/superset/queries/api.py index 3ca8ab87786..3e2f54cb15e 100644 --- a/superset/queries/api.py +++ b/superset/queries/api.py @@ -182,9 +182,8 @@ class QueryRestApi(BaseSupersetModelRestApi): @parse_rison(queries_get_updated_since_schema) @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.get_updated_since" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".get_updated_since", log_to_statsd=False, ) def get_updated_since(self, **kwargs: Any) -> FlaskResponse: diff --git a/superset/reports/api.py b/superset/reports/api.py index 0d03b3972d7..ac8a0582d7b 100644 --- a/superset/reports/api.py +++ b/superset/reports/api.py @@ -621,9 +621,9 @@ class ReportScheduleRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.slack_channels" - ), + action=lambda self, + *args, + **kwargs: f"{self.__class__.__name__}.slack_channels", log_to_statsd=False, ) def slack_channels(self, **kwargs: Any) -> Response: diff --git a/superset/tags/api.py b/superset/tags/api.py index 90809154ebb..04b95f885ce 100644 --- a/superset/tags/api.py +++ b/superset/tags/api.py @@ -617,9 +617,8 @@ class TagRestApi(BaseSupersetModelRestApi): @statsd_metrics @parse_rison({"type": "array", "items": {"type": "integer"}}) @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.favorite_status" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".favorite_status", log_to_statsd=False, ) def favorite_status(self, **kwargs: Any) -> Response: @@ -716,9 +715,8 @@ class TagRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.remove_favorite" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".remove_favorite", log_to_statsd=False, ) def remove_favorite(self, pk: int) -> Response: diff --git a/superset/tasks/api.py b/superset/tasks/api.py index ee390971d13..1870d963b67 100644 --- a/superset/tasks/api.py +++ b/superset/tasks/api.py @@ -380,9 +380,8 @@ class TaskRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.related_subscribers" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + ".related_subscribers", log_to_statsd=False, ) def related_subscribers(self) -> Response: diff --git a/superset/themes/api.py b/superset/themes/api.py index 29cadcabde3..181fc52bbe1 100644 --- a/superset/themes/api.py +++ b/superset/themes/api.py @@ -560,9 +560,9 @@ class ThemeRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.set_system_default" - ), + action=lambda self, + *args, + **kwargs: f"{self.__class__.__name__}.set_system_default", log_to_statsd=False, ) def set_system_default(self, pk: int) -> Response: @@ -627,9 +627,9 @@ class ThemeRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.set_system_dark" - ), + action=lambda self, + *args, + **kwargs: f"{self.__class__.__name__}.set_system_dark", log_to_statsd=False, ) def set_system_dark(self, pk: int) -> Response: @@ -694,9 +694,9 @@ class ThemeRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.unset_system_default" - ), + action=lambda self, + *args, + **kwargs: f"{self.__class__.__name__}.unset_system_default", log_to_statsd=False, ) def unset_system_default(self) -> Response: @@ -744,9 +744,9 @@ class ThemeRestApi(BaseSupersetModelRestApi): @safe @statsd_metrics @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.unset_system_dark" - ), + action=lambda self, + *args, + **kwargs: f"{self.__class__.__name__}.unset_system_dark", log_to_statsd=False, ) def unset_system_dark(self) -> Response: diff --git a/superset/utils/excel.py b/superset/utils/excel.py index 2af29e79a48..e5067417dc8 100644 --- a/superset/utils/excel.py +++ b/superset/utils/excel.py @@ -94,9 +94,9 @@ def apply_column_types( # if the number is too large, convert it to a string # Excel does not support numbers larger than 10^15 df[column] = df[column].apply( - lambda x: ( - str(x) if isinstance(x, (int, float)) and abs(x) > 10**15 else x - ) + lambda x: str(x) + if isinstance(x, (int, float)) and abs(x) > 10**15 + else x ) except ValueError: df[column] = df[column].astype(str) diff --git a/superset/utils/mock_data.py b/superset/utils/mock_data.py index 557547a4bfd..e07583bc276 100644 --- a/superset/utils/mock_data.py +++ b/superset/utils/mock_data.py @@ -122,11 +122,8 @@ def get_type_generator( # pylint: disable=too-many-return-statements,too-many-b sqlalchemy.sql.sqltypes.DateTime, ), ): - return lambda: ( - datetime.fromordinal(MINIMUM_DATE.toordinal()) - + timedelta( - seconds=random.randrange(days_range * 86400) # noqa: S311 - ) + return lambda: datetime.fromordinal(MINIMUM_DATE.toordinal()) + timedelta( + seconds=random.randrange(days_range * 86400) # noqa: S311 ) if isinstance(sqltype, sqlalchemy.sql.sqltypes.Numeric): diff --git a/superset/views/log/api.py b/superset/views/log/api.py index d4f58e8deed..6aaf846416c 100644 --- a/superset/views/log/api.py +++ b/superset/views/log/api.py @@ -108,9 +108,8 @@ class LogRestApi(LogMixin, BaseSupersetModelRestApi): @statsd_metrics @parse_rison(get_recent_activity_schema) @event_logger.log_this_with_context( - action=lambda self, *args, **kwargs: ( - f"{self.__class__.__name__}.recent_activity" - ), + action=lambda self, *args, **kwargs: f"{self.__class__.__name__}" + f".recent_activity", log_to_statsd=False, ) def recent_activity(self, **kwargs: Any) -> FlaskResponse: diff --git a/tests/integration_tests/datasource_tests.py b/tests/integration_tests/datasource_tests.py index 7b8cd24d7e4..ec75ce1f0c9 100644 --- a/tests/integration_tests/datasource_tests.py +++ b/tests/integration_tests/datasource_tests.py @@ -229,8 +229,8 @@ class TestDatasource(SupersetTestCase): def test_external_metadata_by_name_for_virtual_table_uses_mutator(self): self.login(ADMIN_USERNAME) with create_and_cleanup_table() as tbl: - current_app.config["SQL_QUERY_MUTATOR"] = lambda sql, **kwargs: ( - "SELECT 456 as intcol, 'def' as mutated_strcol" + current_app.config["SQL_QUERY_MUTATOR"] = ( + lambda sql, **kwargs: "SELECT 456 as intcol, 'def' as mutated_strcol" ) params = rison.dumps( @@ -365,12 +365,10 @@ class TestDatasource(SupersetTestCase): pytest.raises( SupersetGenericDBErrorException, - lambda: ( - db.session.query(SqlaTable) - .filter_by(id=tbl.id) - .one_or_none() - .external_metadata() - ), + lambda: db.session.query(SqlaTable) + .filter_by(id=tbl.id) + .one_or_none() + .external_metadata(), ) resp = self.client.get(url) diff --git a/tests/integration_tests/db_engine_specs/presto_tests.py b/tests/integration_tests/db_engine_specs/presto_tests.py index da94bdff490..10d9067f75b 100644 --- a/tests/integration_tests/db_engine_specs/presto_tests.py +++ b/tests/integration_tests/db_engine_specs/presto_tests.py @@ -82,8 +82,8 @@ class TestPrestoDbEngineSpec(SupersetTestCase): def verify_presto_column(self, column, expected_results): inspector = mock.Mock() preparer = inspector.engine.dialect.identifier_preparer - preparer.quote_identifier = preparer.quote = preparer.quote_schema = lambda x: ( - f'"{x}"' + preparer.quote_identifier = preparer.quote = preparer.quote_schema = ( + lambda x: f'"{x}"' ) row = mock.Mock() row.Column, row.Type, row.Null = column @@ -828,8 +828,8 @@ class TestPrestoDbEngineSpec(SupersetTestCase): def test_show_columns(self): inspector = mock.MagicMock() preparer = inspector.engine.dialect.identifier_preparer - preparer.quote_identifier = preparer.quote = preparer.quote_schema = lambda x: ( - f'"{x}"' + preparer.quote_identifier = preparer.quote = preparer.quote_schema = ( + lambda x: f'"{x}"' ) inspector.bind.execute.return_value.fetchall = mock.MagicMock( return_value=["a", "b"] @@ -845,8 +845,8 @@ class TestPrestoDbEngineSpec(SupersetTestCase): def test_show_columns_with_schema(self): inspector = mock.MagicMock() preparer = inspector.engine.dialect.identifier_preparer - preparer.quote_identifier = preparer.quote = preparer.quote_schema = lambda x: ( - f'"{x}"' + preparer.quote_identifier = preparer.quote = preparer.quote_schema = ( + lambda x: f'"{x}"' ) inspector.bind.execute.return_value.fetchall = mock.MagicMock( return_value=["a", "b"] diff --git a/tests/integration_tests/security/api_tests.py b/tests/integration_tests/security/api_tests.py index 024d174638f..2b9d4ef0231 100644 --- a/tests/integration_tests/security/api_tests.py +++ b/tests/integration_tests/security/api_tests.py @@ -260,9 +260,8 @@ class TestSecurityGuestTokenApiTokenValidator(SupersetTestCase): @with_config( { - "GUEST_TOKEN_VALIDATOR_HOOK": lambda x: ( - len(x["rls"]) == 1 and "tenant_id=" in x["rls"][0]["clause"] - ) + "GUEST_TOKEN_VALIDATOR_HOOK": lambda x: len(x["rls"]) == 1 + and "tenant_id=" in x["rls"][0]["clause"] } ) def test_guest_validator_hook_real_world_example_positive(self): @@ -277,9 +276,8 @@ class TestSecurityGuestTokenApiTokenValidator(SupersetTestCase): @with_config( { - "GUEST_TOKEN_VALIDATOR_HOOK": lambda x: ( - len(x["rls"]) == 1 and "tenant_id=" in x["rls"][0]["clause"] - ) + "GUEST_TOKEN_VALIDATOR_HOOK": lambda x: len(x["rls"]) == 1 + and "tenant_id=" in x["rls"][0]["clause"] } ) def test_guest_validator_hook_real_world_example_negative(self): diff --git a/tests/unit_tests/common/test_query_context_factory.py b/tests/unit_tests/common/test_query_context_factory.py index 4c37ba42f17..37a7b47460d 100644 --- a/tests/unit_tests/common/test_query_context_factory.py +++ b/tests/unit_tests/common/test_query_context_factory.py @@ -140,8 +140,8 @@ class TestQueryContextFactory: existing_columns = {"existing_col"} with patch.object(self.factory, "_find_column_definition") as mock_find: - mock_find.side_effect = lambda qo, col: ( - f"def_{col}" if col != "tooltip_col2" else None + mock_find.side_effect = ( + lambda qo, col: f"def_{col}" if col != "tooltip_col2" else None ) self.factory._append_missing_tooltip_columns( diff --git a/tests/unit_tests/datasets/test_datetime_format_detector.py b/tests/unit_tests/datasets/test_datetime_format_detector.py index 72f63f2bb41..d66e2efe8e2 100644 --- a/tests/unit_tests/datasets/test_datetime_format_detector.py +++ b/tests/unit_tests/datasets/test_datetime_format_detector.py @@ -44,8 +44,8 @@ def mock_dataset() -> MagicMock: dataset.database.get_sqla_engine.return_value.__exit__.return_value = None # Mock apply_limit_to_sql to return SQL with LIMIT - dataset.database.apply_limit_to_sql = lambda sql, limit, force: ( - f"{sql} LIMIT {limit}" + dataset.database.apply_limit_to_sql = ( + lambda sql, limit, force: f"{sql} LIMIT {limit}" ) return dataset diff --git a/tests/unit_tests/db_engine_specs/test_starrocks.py b/tests/unit_tests/db_engine_specs/test_starrocks.py index ca721d608c7..e3ef836f36d 100644 --- a/tests/unit_tests/db_engine_specs/test_starrocks.py +++ b/tests/unit_tests/db_engine_specs/test_starrocks.py @@ -230,8 +230,8 @@ def test_get_catalog_names(mocker: MockerFixture) -> None: # StarRocks returns rows with keys: ['Catalog', 'Type', 'Comment'] mock_row_1 = mocker.MagicMock() mock_row_1.keys.return_value = ["Catalog", "Type", "Comment"] - mock_row_1.__getitem__ = lambda self, key: ( - "default_catalog" if key == "Catalog" else None + mock_row_1.__getitem__ = ( + lambda self, key: "default_catalog" if key == "Catalog" else None ) mock_row_2 = mocker.MagicMock() diff --git a/tests/unit_tests/security/guest_rls_test.py b/tests/unit_tests/security/guest_rls_test.py index 56177bb4b0f..6210570d66f 100644 --- a/tests/unit_tests/security/guest_rls_test.py +++ b/tests/unit_tests/security/guest_rls_test.py @@ -192,8 +192,8 @@ def _make_datasource_with_real_rls(dataset_id: int) -> MagicMock: datasource = _make_datasource(dataset_id) # Bind real BaseDatasource method so RLS logic executes against mocked # security_manager rather than returning MagicMock auto-stub - datasource.get_sqla_row_level_filters = lambda **kwargs: ( - BaseDatasource.get_sqla_row_level_filters(datasource, **kwargs) + datasource.get_sqla_row_level_filters = ( + lambda **kwargs: BaseDatasource.get_sqla_row_level_filters(datasource, **kwargs) ) return datasource diff --git a/tests/unit_tests/test_viz_cache_key.py b/tests/unit_tests/test_viz_cache_key.py index 17de2f5f404..d6eb614265c 100644 --- a/tests/unit_tests/test_viz_cache_key.py +++ b/tests/unit_tests/test_viz_cache_key.py @@ -97,8 +97,8 @@ def test_cache_query_by_user_flag_yields_distinct_keys(feature_flag_mock): """ Global ``CACHE_QUERY_BY_USER`` flag also reaches the legacy viz path. """ - feature_flag_mock.is_feature_enabled.side_effect = lambda feature=None: ( - feature == "CACHE_QUERY_BY_USER" + feature_flag_mock.is_feature_enabled.side_effect = ( + lambda feature=None: feature == "CACHE_QUERY_BY_USER" ) database = Database(database_name="d", sqlalchemy_uri="sqlite://") obj = _viz_for(database) diff --git a/tests/unit_tests/utils/test_core.py b/tests/unit_tests/utils/test_core.py index d40d19aca09..2b852796ebf 100644 --- a/tests/unit_tests/utils/test_core.py +++ b/tests/unit_tests/utils/test_core.py @@ -672,9 +672,8 @@ def test_get_user_agent(mocker: MockerFixture, app_context: None) -> None: @with_config( { - "USER_AGENT_FUNC": lambda database, source: ( - f"{database.database_name} {source.name}" - ) + "USER_AGENT_FUNC": lambda database, + source: f"{database.database_name} {source.name}" } ) def test_get_user_agent_custom(mocker: MockerFixture, app_context: None) -> None: