black formatting

Author: cvanelteren

ultraplot/axes/plot.py

@@ -2479,15 +2479,15 @@ def _parse_cycle(
         """
         cycle_kw = cycle_kw or {}
         cycle_manually = cycle_manually or {}
-        cycle_kw.setdefault('N', ncycle)
+        cycle_kw.setdefault("N", ncycle)
 
         # Match-case for cycle resolution
         match cycle:
             case None if not cycle_kw:
                 resolved_cycle = None
             case True:
-                resolved_cycle = constructor.Cycle(rc['axes.prop_cycle'])
-            case str() if cycle.lower() == 'none':
+                resolved_cycle = constructor.Cycle(rc["axes.prop_cycle"])
+            case str() if cycle.lower() == "none":
                 resolved_cycle = None
             case str() | int():
                 resolved_cycle = constructor.Cycle(cycle, **cycle_kw)
@@ -2507,16 +2507,15 @@ def _parse_cycle(
         # Apply manual cycle properties
         if cycle_manually:
             current_prop = self._get_lines._cycler_items[self._get_lines._idx]
-            self._get_lines._idx = (self._get_lines._idx + 1) % len(self._get_lines._cycler_items)
+            self._get_lines._idx = (self._get_lines._idx + 1) % len(
+                self._get_lines._cycler_items
+            )
             for prop, key in cycle_manually.items():
                 if kwargs.get(key) is None and prop in current_prop:
                     value = current_prop[prop]
-                    kwargs[key] = pcolors.to_hex(value) if key == 'c' else value
+                    kwargs[key] = pcolors.to_hex(value) if key == "c" else value
         return kwargs
 
-
-
-
     def _parse_level_lim(
         self,
         *args,

ultraplot/constructor.py

@@ -155,9 +155,7 @@
     "lat": partial(pticker.SimpleFormatter, negpos="SN"),
     "lon": partial(pticker.SimpleFormatter, negpos="WE", wraprange=(-180, 180)),
     "deg": partial(pticker.SimpleFormatter, suffix="\N{DEGREE SIGN}"),
-    "deglat": partial(
-        pticker.SimpleFormatter, suffix="\N{DEGREE SIGN}", negpos="SN"
-    ),
+    "deglat": partial(pticker.SimpleFormatter, suffix="\N{DEGREE SIGN}", negpos="SN"),
     "deglon": partial(
         pticker.SimpleFormatter,
         suffix="\N{DEGREE SIGN}",
@@ -862,9 +860,7 @@ def __init__(self, *args, N=None, samples=None, name=None, **kwargs):
         elif self._is_all_cyclers(args):
             self._handle_cycler_args(args, cycler_props, kwargs)
         else:
-            self._handle_colormap_args(
-                args, cycler_props, kwargs, samples, name
-            )
+            self._handle_colormap_args(args, cycler_props, kwargs, samples, name)
 
         self._iterator = None  # internal reference for cycle
 
@@ -887,17 +883,13 @@ def _handle_empty_args(self, props, kwargs):
         """Handle case when no positional arguments are provided."""
         props.setdefault("color", "black")
         if kwargs:
-            warnings._warn_ultraplot(
-                f"Ignoring Cycle() keyword arg(s) {kwargs}."
-            )
+            warnings._warn_ultraplot(f"Ignoring Cycle() keyword arg(s) {kwargs}.")
         self._build_cycler(())
 
     def _handle_cycler_args(self, args, props, kwargs):
         """Handle case when arguments are cycler objects."""
         if kwargs:
-            warnings._warn_ultraplot(
-                f"Ignoring Cycle() keyword arg(s) {kwargs}."
-            )
+            warnings._warn_ultraplot(f"Ignoring Cycle() keyword arg(s) {kwargs}.")
         if len(args) == 1 and not props:
             self._build_cycler((args[0].by_key(),))
         else:
@@ -912,11 +904,7 @@ def _handle_colormap_args(self, args, props, kwargs, samples, name):
             )
 
         cmap = self._create_colormap(args, name, samples, kwargs)
-        dict_ = {
-            "color": [
-                c if isinstance(c, str) else to_hex(c) for c in cmap.colors
-            ]
-        }
+        dict_ = {"color": [c if isinstance(c, str) else to_hex(c) for c in cmap.colors]}
         self._build_cycler((dict_, props))
         self.name = _not_none(name, cmap.name)
 
@@ -946,9 +934,7 @@ def _build_cycler(self, dicts):
         # Build cycler with matching property lengths
         lengths = [len(value) for value in props.values()]
         maxlen = np.lcm.reduce(lengths)
-        props = {
-            key: value * (maxlen // len(value)) for key, value in props.items()
-        }
+        props = {key: value * (maxlen // len(value)) for key, value in props.items()}
         mcycler = cycler.cycler(**props)
         super().__init__(mcycler)
 
@@ -1286,9 +1272,7 @@ def Formatter(formatter, *args, date=False, index=False, **kwargs):
     elif formatter is False:
         formatter = mticker.NullFormatter(*args, **kwargs)
     elif np.iterable(formatter):
-        formatter = (mticker.FixedFormatter, pticker.IndexFormatter)[index](
-            formatter
-        )
+        formatter = (mticker.FixedFormatter, pticker.IndexFormatter)[index](formatter)
     elif callable(formatter):
         formatter = mticker.FuncFormatter(formatter, *args, **kwargs)
     else:
@@ -1594,9 +1578,7 @@ def Proj(
         if backend is not None:
             kwargs["backend"] = backend
         if kwargs:
-            warnings._warn_ultraplot(
-                f"Ignoring Proj() keyword arg(s): {kwargs!r}."
-            )
+            warnings._warn_ultraplot(f"Ignoring Proj() keyword arg(s): {kwargs!r}.")
         proj = name
         backend = "cartopy" if is_crs else "basemap"
 
@@ -1621,19 +1603,14 @@ def Proj(
             message = f"{name!r} is an unknown cartopy projection class.\n"
             message += "The known cartopy projection classes are:\n"
             message += "\n".join(
-                " "
-                + key
-                + " " * (max(map(len, PROJS)) - len(key) + 10)
-                + cls.__name__
+                " " + key + " " * (max(map(len, PROJS)) - len(key) + 10) + cls.__name__
                 for key, cls in PROJS.items()
             )
             if include_axes:
                 from . import axes as paxes  # avoid circular imports
 
                 message = message.replace("class.", "class or axes subclass.")
-                message += (
-                    "\nThe known axes subclasses are:\n" + paxes._cls_table
-                )
+                message += "\nThe known axes subclasses are:\n" + paxes._cls_table
             raise ValueError(message) from None
         if name == "geos":  # fix common mistake
             kwargs.pop("central_latitude", None)
@@ -1687,12 +1664,8 @@ def Proj(
             if include_axes:
                 from . import axes as paxes  # avoid circular imports
 
-                message = message.replace(
-                    "projection.", "projection or axes subclass."
-                )
-                message += (
-                    "\nThe known axes subclasses are:\n" + paxes._cls_table
-                )
+                message = message.replace("projection.", "projection or axes subclass.")
+                message += "\nThe known axes subclasses are:\n" + paxes._cls_table
             raise ValueError(message) from None
 
     proj._proj_backend = backend

ultraplot/tests/test_1dplots.py

@@ -303,9 +303,7 @@ def test_scatter_inbounds():
     N = 100
     fig.format(xlim=(0, 20))
     for i, ax in enumerate(axs):
-        c = ax.scatter(
-            np.arange(N), np.arange(N), c=np.arange(N), inbounds=bool(i)
-        )
+        c = ax.scatter(np.arange(N), np.arange(N), c=np.arange(N), inbounds=bool(i))
         ax.colorbar(c, loc="b")
     return fig
 
@@ -321,9 +319,7 @@ def test_scatter_alpha():
     ax.scatter(data, alpha=alpha)
     ax.scatter(data + 1, c=np.arange(data.size), cmap="BuRd", alpha=alpha)
     ax.scatter(data + 2, color="k", alpha=alpha)
-    ax.scatter(
-        data + 3, color=[f"red{i}" for i in range(data.size)], alpha=alpha
-    )
+    ax.scatter(data + 3, color=[f"red{i}" for i in range(data.size)], alpha=alpha)
     return fig
 
 
@@ -362,9 +358,7 @@ def test_scatter_sizes():
             kw = {"absolute_size": i == 2}
             if i == 1:
                 kw["smin"] = 0
-                kw["smax"] = (
-                    size**2
-                )  # should be same as relying on lines.markersize
+                kw["smax"] = size**2  # should be same as relying on lines.markersize
             ax.scatter(np.arange(5), [0.25 * (1 + i)] * 5, size**2, **kw)
     # Test various size arguments
     ax = fig.subplot(122, margin=0.15)
@@ -377,12 +371,8 @@ def test_scatter_sizes():
         area_size=False,
         absolute_size=True,
     )
-    ax.scatter(
-        np.arange(5), [0.50] * 5, c="red7", sizes=data, absolute_size=True
-    )
-    ax.scatter(
-        np.arange(5), [0.75] * 5, c="red7", sizes=data, absolute_size=False
-    )
+    ax.scatter(np.arange(5), [0.50] * 5, c="red7", sizes=data, absolute_size=True)
+    ax.scatter(np.arange(5), [0.75] * 5, c="red7", sizes=data, absolute_size=False)
     for i, d in enumerate(data):
         ax.text(i, 0.5, format(d, ".0f"), va="center", ha="center")
     return fig