MatplotLibAPI 3.2.19__tar.gz → 3.2.20__tar.gz

{matplotlibapi-3.2.19 → matplotlibapi-3.2.20}/.github/workflows/ci.yml

@@ -31,12 +31,12 @@ jobs:
 
     - name: Run style checks
       run: |
-        pydocstyle MatplotLibAPI
+        pydocstyle .
         black --check .
 
     - name: Run static type analysis
       run: |
-        pyright MatplotLibAPI
+        pyright .
 
     - name: Run tests with coverage
       run: |

{matplotlibapi-3.2.19 → matplotlibapi-3.2.20}/MatplotLibAPI/BoxViolin.py

@@ -33,10 +33,12 @@ def aplot_box_violin(
     validate_dataframe(pd_df, cols=cols)
     plot_ax = _get_axis(ax)
 
+    common_kwargs = {"data": pd_df, "x": by, "y": column, "palette": style.palette}
+
     if violin:
-        sns.violinplot(data=pd_df, x=by, y=column, palette=style.palette, ax=plot_ax)
+        sns.violinplot(**common_kwargs, hue=by, legend=False, ax=plot_ax)
     else:
-        sns.boxplot(data=pd_df, x=by, y=column, palette=style.palette, ax=plot_ax)
+        sns.boxplot(**common_kwargs, hue=by, legend=False, ax=plot_ax)
 
     plot_ax.set_facecolor(style.background_color)
     plot_ax.set_ylabel(string_formatter(column))

{matplotlibapi-3.2.19 → matplotlibapi-3.2.20}/MatplotLibAPI/Wordcloud.py

@@ -25,33 +25,6 @@ WORDCLOUD_STYLE_TEMPLATE = StyleTemplate(
 )
 
 
-def _normalize_weights(weights: Sequence[float], base_size: int) -> np.ndarray:
-    """Normalize weights to a range of font sizes.
-
-    Parameters
-    ----------
-    weights : Sequence[float]
-        Sequence of weights representing word importance.
-    base_size : int
-        Base font size used as the lower bound for scaling.
-
-    Returns
-    -------
-    numpy.ndarray
-        Array of font sizes corresponding to the provided weights.
-    """
-    numeric_weights = np.asarray(weights, dtype=float)
-    if numeric_weights.size == 0:
-        return np.array([], dtype=float)
-    min_weight = numeric_weights.min()
-    max_weight = numeric_weights.max()
-    if min_weight == max_weight:
-        return np.full_like(numeric_weights, fill_value=base_size, dtype=float)
-
-    min_size, max_size = base_size, base_size * 4
-    return np.interp(numeric_weights, (min_weight, max_weight), (min_size, max_size))
-
-
 def _filter_stopwords(
     words: Iterable[str], stopwords: Optional[Iterable[str]]
 ) -> np.ndarray:
@@ -232,20 +205,21 @@ def _plot_words(
     frequency_map = {
         string_formatter(word): weight for word, weight in zip(words, weights)
     }
+    min_font_size = style.font_mapping[min(style.font_mapping.keys())]
+    max_font_size = style.font_mapping[max(style.font_mapping.keys())]
 
-    font_sizes = _normalize_weights(weights, base_size=style.font_size)
     wc = WordCloud(
         width=width,
         height=height,
         background_color=style.background_color,
         colormap=colormaps.get_cmap(style.palette),
-        min_font_size=int(font_sizes.min(initial=style.font_size)),
-        max_font_size=int(font_sizes.max(initial=style.font_size * 4)),
+        min_font_size=min_font_size,
+        max_font_size=max_font_size,
         random_state=random_state,
         mask=resolved_mask,
     ).generate_from_frequencies(frequency_map)
 
-    ax.imshow(wc, interpolation="bilinear")
+    ax.imshow(wc.to_array(), interpolation="bilinear")
 
     if title:
         ax.set_title(title, color=style.font_color, fontsize=style.font_size * 1.5)
@@ -261,7 +235,7 @@ def aplot_wordcloud(
     max_words: int = MAX_RESULTS,
     stopwords: Optional[Iterable[str]] = None,
     random_state: Optional[int] = None,
-    ax: Optional[Axes] = None,
+    ax: Optional[Axes | np.ndarray[Any, np.dtype[Any]]] = None,
     mask: Optional[np.ndarray] = None,
 ) -> Axes:
     """Plot a word cloud on the provided axes.
@@ -284,7 +258,7 @@ def aplot_wordcloud(
         Words to exclude from the visualization. Defaults to ``None``.
     random_state : int, optional
         Seed for word placement. Defaults to ``None``.
-    ax : matplotlib.axes.Axes, optional
+    ax : matplotlib.axes.Axes or numpy.ndarray, optional
         Axes to draw on. Defaults to ``None`` which uses the current axes.
     mask : numpy.ndarray, optional
         Two-dimensional mask array defining the drawable region of the word cloud.
@@ -299,9 +273,15 @@ def aplot_wordcloud(
     ------
     AttributeError
         If required columns are missing from the DataFrame.
+    TypeError
+        If ``ax`` is not a ``matplotlib.axes.Axes`` instance.
     """
     if ax is None:
         ax = cast(Axes, plt.gca())
+    elif isinstance(ax, np.ndarray):
+        raise TypeError("ax must be a single matplotlib Axes instance, not an array.")
+    elif not isinstance(ax, Axes):
+        raise TypeError("ax must be a matplotlib Axes instance.")
 
     words, weights = _prepare_word_frequencies(
         pd_df=pd_df,

{matplotlibapi-3.2.19 → matplotlibapi-3.2.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: MatplotLibAPI
-Version: 3.2.19
+Version: 3.2.20
 License-File: LICENSE
 Requires-Python: >=3.8
 Requires-Dist: kaleido

{matplotlibapi-3.2.19 → matplotlibapi-3.2.20}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["hatchling"]
 build-backend = "hatchling.build"
 [project]
 name = "MatplotLibAPI"
-version = "3.2.19"
+version = "3.2.20"
 readme = "README.md"
 requires-python = ">=3.8"
 dependencies = [

{matplotlibapi-3.2.19 → matplotlibapi-3.2.20}/scripts/generate_sample_data.py

@@ -97,7 +97,7 @@ def generate_treemap_data():
     }
     df = pd.DataFrame(data)
     # For treemap, we need a path-like structure. We will create it here.
-    df["path"] = df["parent"] + "/" + df["location"]
+    df["path"] = df["parent"].str.cat(df["location"], sep="/")
     df["path"] = df["path"].str.lstrip("/")
     df.to_csv("data/treemap.csv", index=False)
 
@@ -169,7 +169,6 @@ def generate_wordcloud_data():
 
 def generate_bar_data():
     """Generate and save sample data for bar and stacked bar charts."""
-
     data = {
         "product": ["Gadget", "Gadget", "Gadget", "Widget", "Widget", "Widget"],
         "region": ["North", "South", "West", "North", "South", "West"],
@@ -181,7 +180,6 @@ def generate_bar_data():
 
 def generate_histogram_data():
     """Generate and save sample data for histogram and KDE plots."""
-
     data = {
         "waiting_time_minutes": [
             5,
@@ -212,7 +210,6 @@ def generate_histogram_data():
 
 def generate_box_violin_data():
     """Generate and save sample data for box and violin plots."""
-
     data = {
         "department": [
             "Engineering",
@@ -243,7 +240,6 @@ def generate_box_violin_data():
 
 def generate_heatmap_and_correlation_data():
     """Generate and save sample data for heatmaps and correlation matrices."""
-
     data = {
         "month": ["Jan", "Jan", "Feb", "Feb", "Mar", "Mar", "Apr", "Apr"],
         "channel": [
@@ -269,7 +265,6 @@ def generate_heatmap_and_correlation_data():
 
 def generate_area_data():
     """Generate and save sample data for area charts."""
-
     data = {
         "quarter": ["Q1", "Q2", "Q3", "Q4", "Q1", "Q2", "Q3", "Q4"],
         "segment": [
@@ -290,7 +285,6 @@ def generate_area_data():
 
 def generate_pie_waffle_data():
     """Generate and save sample data for pie, donut, and waffle charts."""
-
     data = {
         "device": ["Desktop", "Mobile", "Tablet", "Other"],
         "sessions": [5200, 8900, 1300, 600],
@@ -302,7 +296,6 @@ def generate_pie_waffle_data():
 
 def generate_sankey_data():
     """Generate and save sample data for Sankey diagrams."""
-
     data = {
         "source": ["Homepage", "Homepage", "Landing Page", "Landing Page", "Cart"],
         "target": ["Landing Page", "Product", "Product", "Signup", "Checkout"],

matplotlibapi-3.2.20/tests/__init__.py

@@ -0,0 +1 @@
+"""Public test for MatplotLibAPI."""

{matplotlibapi-3.2.19 → matplotlibapi-3.2.20}/tests/conftest.py

@@ -2,7 +2,7 @@
 
 import os
 from pathlib import Path
-from typing import Any, Callable
+from typing import Any, Callable, Generator
 
 import pandas as pd
 import pytest
@@ -13,7 +13,6 @@ from scripts import generate_sample_data
 @pytest.fixture(scope="session")
 def sample_data_dir(tmp_path_factory: pytest.TempPathFactory) -> Path:
     """Generate sample CSV files in an isolated directory for tests."""
-
     base_dir = tmp_path_factory.mktemp("sample_data")
     cwd = os.getcwd()
     os.chdir(base_dir)
@@ -28,7 +27,7 @@ def sample_data_dir(tmp_path_factory: pytest.TempPathFactory) -> Path:
 @pytest.fixture(scope="session")
 def load_sample_df(
     sample_data_dir: Path,
-) -> Callable[[str, Any], pd.DataFrame]:
+) -> Callable[..., pd.DataFrame]:
     """Return a loader that reads generated sample CSVs into dataframes."""
 
     def _loader(filename: str, **kwargs: Any) -> pd.DataFrame:
@@ -38,9 +37,8 @@ def load_sample_df(
 
 
 @pytest.fixture(autouse=True)
-def close_matplotlib_figures() -> None:
+def close_matplotlib_figures() -> Generator[None, None, None]:
     """Close any matplotlib figures created during a test."""
-
     yield
 
     import matplotlib.pyplot as plt

matplotlibapi-3.2.20/tests/test_wordcloud.py

@@ -0,0 +1,77 @@
+"""Tests for word cloud visualizations."""
+
+import matplotlib.pyplot as plt
+from matplotlib.figure import Figure
+import numpy as np
+
+from MatplotLibAPI.Wordcloud import (
+    aplot_wordcloud,
+    create_circular_mask,
+    fplot_wordcloud,
+)
+
+
+def test_fplot_wordcloud(load_sample_df):
+    """Render a word cloud figure from sample data."""
+
+    df = load_sample_df("wordcloud.csv")
+
+    fig = fplot_wordcloud(
+        pd_df=df, text_column="country", weight_column="population", random_state=42
+    )
+
+    assert isinstance(fig, Figure)
+    default_mask = create_circular_mask(size=300)
+    image = fig.axes[0].images[0].get_array()
+    assert image is not None
+    assert tuple(image.shape[:2]) == default_mask.shape
+
+
+def test_fplot_wordcloud_with_mask(load_sample_df):
+    """Render a word cloud using a circular mask to constrain placement."""
+
+    df = load_sample_df("wordcloud.csv")
+    mask = create_circular_mask(size=200)
+
+    fig = fplot_wordcloud(
+        pd_df=df,
+        text_column="country",
+        weight_column="population",
+        random_state=0,
+        mask=mask,
+    )
+
+    image = fig.axes[0].images[0].get_array()
+    assert image is not None
+    assert tuple(image.shape[:2]) == mask.shape
+
+
+def test_aplot_wordcloud(load_sample_df):
+    """Render a word cloud onto an existing axes object."""
+
+    df = load_sample_df("wordcloud.csv")
+    fig, ax = plt.subplots()
+
+    result_ax = aplot_wordcloud(
+        ax=ax,
+        pd_df=df,
+        text_column="country",
+        weight_column="population",
+        random_state=42,
+    )
+
+    assert result_ax is not None
+    assert ax is result_ax
+    image = result_ax.images[0].get_array()
+    assert image is not None
+    assert image.shape[0] > 0 and image.shape[1] > 0
+
+
+def test_create_circular_mask():
+    """Verify circular mask generation."""
+
+    mask = create_circular_mask(size=100, radius=40)
+    assert mask.shape == (100, 100)
+    assert mask.dtype == np.uint8
+    assert np.sum(mask == 0) > 0
+    assert np.sum(mask == 255) > 0

matplotlibapi-3.2.19/tests/test_wordcloud.py

@@ -1,38 +0,0 @@
-"""Tests for word cloud visualizations."""
-
-from matplotlib.figure import Figure
-
-from MatplotLibAPI.Wordcloud import create_circular_mask, fplot_wordcloud
-
-
-def test_fplot_wordcloud(load_sample_df):
-    """Render a word cloud figure from sample data."""
-
-    df = load_sample_df("wordcloud.csv")
-
-    fig = fplot_wordcloud(
-        pd_df=df, text_column="country", weight_column="population", random_state=42
-    )
-
-    assert isinstance(fig, Figure)
-    default_mask = create_circular_mask()
-    image = fig.axes[0].images[0].get_array()
-    assert tuple(image.shape[:2]) == default_mask.shape
-
-
-def test_fplot_wordcloud_with_mask(load_sample_df):
-    """Render a word cloud using a circular mask to constrain placement."""
-
-    df = load_sample_df("wordcloud.csv")
-    mask = create_circular_mask(size=200)
-
-    fig = fplot_wordcloud(
-        pd_df=df,
-        text_column="country",
-        weight_column="population",
-        random_state=0,
-        mask=mask,
-    )
-
-    image = fig.axes[0].images[0].get_array()
-    assert tuple(image.shape[:2]) == mask.shape