style

Author: fresioAS

sqlmesh/core/engine_adapter/fabric.py

@@ -61,22 +61,23 @@ def _connected_catalog(self) -> t.Optional[str]:
     def _connected_catalog(self, value: t.Optional[str]) -> None:
         self._connection_pool.set_attribute("connected_catalog", value)
 
-    def _normalize_catalog(
-        self, catalog_name: t.Optional[str]
-    ) -> t.Optional[str]:
+    def _normalize_catalog(self, catalog_name: t.Optional[str]) -> t.Optional[str]:
         if not catalog_name:
             return None
 
-        default_catalog = (
-            self._default_catalog or self._extra_config.get("database")
-        )
+        default_catalog = self._default_catalog or self._extra_config.get("database")
         if default_catalog and catalog_name == default_catalog:
             return None
 
         return catalog_name
 
     def _catalog_state_label(self, catalog_name: t.Optional[str]) -> str:
-        return catalog_name or self._default_catalog or self._extra_config.get("database") or "<default>"
+        return (
+            catalog_name
+            or self._default_catalog
+            or self._extra_config.get("database")
+            or "<default>"
+        )
 
     @property
     def api_client(self) -> FabricHttpClient:
@@ -166,9 +167,7 @@ def set_current_catalog(self, catalog_name: t.Optional[str]) -> None:
 
         # No-op: the logical catalog state already matches.
         if self.get_current_catalog() == target_catalog:
-            logger.debug(
-                "Already using requested Fabric catalog state, no action needed"
-            )
+            logger.debug("Already using requested Fabric catalog state, no action needed")
             return
 
         # Decide whether the open connection needs to be replaced.
@@ -188,9 +187,7 @@ def set_current_catalog(self, catalog_name: t.Optional[str]) -> None:
         #    restore-to-neutral left the connection on the right catalog, we
         #    skip the close entirely.
         connected_catalog = self._normalize_catalog(self._connected_catalog)
-        needs_reconnect = (
-            target_catalog is not None and connected_catalog != target_catalog
-        )
+        needs_reconnect = target_catalog is not None and connected_catalog != target_catalog
 
         if needs_reconnect:
             logger.info(

tests/core/engine_adapter/test_fabric.py

@@ -137,8 +137,8 @@ def test_switching_between_catalogs_closes_each_time(
     close_spy = mocker.spy(adapter._connection_pool, "close")
     adapter.cursor.fetchone.return_value = (1,)
 
-    adapter.table_exists("safran.db.table")   # None→safran: 1 close
-    adapter.table_exists("planning.db.table") # safran→planning: 2nd close
+    adapter.table_exists("safran.db.table")  # None→safran: 1 close
+    adapter.table_exists("planning.db.table")  # safran→planning: 2nd close
 
     assert close_spy.call_count == 2
     assert adapter._connected_catalog == "planning"