tesorotools-python 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

tesorotools/artists/_common.py

@@ -24,6 +24,7 @@ overrides only need to update ``plots.yaml``.
 from __future__ import annotations
 
 import math
+import warnings
 from collections.abc import Sequence
 from pathlib import Path
 from typing import Any, Literal, Protocol, Self, cast, runtime_checkable
@@ -624,7 +625,7 @@ def compact_legend_font_pt() -> float:
     )
 
 
-def compact_legend_kwargs() -> dict[str, Any]:
+def compact_legend_kwargs(ax: Axes | None = None) -> dict[str, Any]:
     """Legend styling for compact charts, from the ``compact`` config.
 
     Returns ``fontsize`` / ``handlelength`` / ``handletextpad`` /
@@ -633,9 +634,17 @@ def compact_legend_kwargs() -> dict[str, Any]:
     the metrics used to *draw* the legend in :func:`legend_above` -- the
     invariant that prevents an over-counted ``ncol``.  The smaller font
     and shorter handles let several wide labels share one legend row.
+
+    With *ax*, the legend is never larger than its tick labels: a chart
+    drawn with a small ``font_pt`` (the 7 pt of the notes) keeps the
+    legend at that size instead of ``legend_font_pt``.
     """
+    fontsize = compact_legend_font_pt()
+    ticks = (ax.get_yticklabels() or ax.get_xticklabels()) if ax else []
+    if ticks:
+        fontsize = min(fontsize, float(ticks[0].get_fontsize()))
     return {
-        "fontsize": compact_legend_font_pt(),
+        "fontsize": fontsize,
         "handlelength": float(COMPACT_CONFIG.get("legend_handlelength", 1.2)),
         "handletextpad": float(COMPACT_CONFIG.get("legend_handletextpad", 0.4)),
         "columnspacing": float(COMPACT_CONFIG.get("legend_columnspacing", 0.8)),
@@ -690,14 +699,21 @@ def date_span_ticks(
     lo: float = float(cast(Any, date2num(data_min)))
     hi: float = float(cast(Any, date2num(data_max)))
     locator = AutoDateLocator(minticks=2, maxticks=maxticks)
-    candidates: list[float] = cast(
-        "list[float]", locator.tick_values(data_min, data_max)
-    )
-    if not pin_start:
-        return [t for t in candidates if lo <= t <= hi] or [lo, hi]
-    chosen: Any = locator.get_locator(  # type: ignore[reportUnknownMemberType]
-        data_min, data_max
-    )
+    # With few ticks over a long span the locator finds no interval in its
+    # table, warns and falls back to a sensible default: the fallback is
+    # the wanted result, the warning only noise for whoever draws a chart.
+    with warnings.catch_warnings():
+        warnings.filterwarnings(
+            "ignore", "AutoDateLocator was unable", UserWarning
+        )
+        candidates: list[float] = cast(
+            "list[float]", locator.tick_values(data_min, data_max)
+        )
+        if not pin_start:
+            return [t for t in candidates if lo <= t <= hi] or [lo, hi]
+        chosen: Any = locator.get_locator(  # type: ignore[reportUnknownMemberType]
+            data_min, data_max
+        )
     if isinstance(chosen, YearLocator):
         step = int(cast(Any, chosen.base).step)
         if step > 1:
@@ -1251,7 +1267,7 @@ def place_legend(
     side = resolve_legend_position(position, compact=compact)
     fig_w_px: float = fig.get_size_inches()[0] * fig.dpi
     if compact:
-        legend_kw = compact_legend_kwargs()
+        legend_kw = compact_legend_kwargs(ax)
         if ncol is None:
             ncol = auto_ncol(
                 ax, labels, available_width_px=fig_w_px, **legend_kw

tesorotools/artists/box_plot.py

@@ -499,7 +499,7 @@ class BoxPlot(FigureLifecycle):
             self.legend.position, compact=self.compact
         )
         if self.compact:
-            kw = compact_legend_kwargs()
+            kw = compact_legend_kwargs(ax)
             ncol = self.legend.ncol or auto_ncol(
                 ax, labels, available_width_px=fig_w_px, **kw
             )

tesorotools/note/__init__.py

@@ -15,26 +15,51 @@ from tesorotools.note.builder import (
     Chart,
     Note,
     Table,
-    chart_size,
     format_number,
+    normalize_unit,
     spanish_date,
 )
-from tesorotools.note.lint import Issue, check
-from tesorotools.note.markdown import NoteSource, build, parse, read
+from tesorotools.note.charts import (
+    NOTE_COLORS,
+    axis_decimals,
+    axis_format,
+    chart_size,
+    note_palette,
+)
+from tesorotools.note.cli import skill_text
+from tesorotools.note.issues import Issue, NoteError
+from tesorotools.note.lint import check
+from tesorotools.note.markdown import (
+    ChartRenderer,
+    NoteSource,
+    assemble,
+    build,
+    parse,
+    read,
+)
 from tesorotools.note.template import open_template
 
 __all__ = [
+    "NOTE_COLORS",
     "Chart",
+    "ChartRenderer",
     "Issue",
     "Note",
+    "NoteError",
     "NoteSource",
     "Table",
+    "assemble",
+    "axis_decimals",
+    "axis_format",
     "build",
     "chart_size",
     "check",
     "format_number",
+    "normalize_unit",
+    "note_palette",
     "open_template",
     "parse",
     "read",
+    "skill_text",
     "spanish_date",
 ]