plot3 0.6.1__tar.gz → 0.6.3__tar.gz

{plot3-0.6.1 → plot3-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plot3
-Version: 0.6.1
+Version: 0.6.3
 Summary: ggplot2 grammar of graphics for Python: interactive WebGL figures and journal-ready PNG, SVG, and PDF
 Author: Rigoberto Leyva Salmeron
 License-Expression: MIT
@@ -17,18 +17,12 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy>=1.24
 Requires-Dist: pandas>=2.0
-Provides-Extra: fast
-Requires-Dist: contourpy>=1.0; extra == "fast"
+Provides-Extra: pdf
+Requires-Dist: cairosvg>=2.7; extra == "pdf"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0; extra == "dev"
 Requires-Dist: polars>=1.0; extra == "dev"
 Requires-Dist: contourpy>=1.0; extra == "dev"
-Provides-Extra: jupyter
-Requires-Dist: ipython>=8.0; extra == "jupyter"
-Provides-Extra: polars
-Requires-Dist: polars>=1.0; extra == "polars"
-Provides-Extra: export
-Requires-Dist: cairosvg>=2.7; extra == "export"
 Dynamic: license-file
 
 # plot3
@@ -57,7 +51,7 @@ in the notebook and save as journal-ready PNG, SVG, or PDF.
 </table>
 
 ```bash
-pip install "plot3[jupyter,export]"
+pip install plot3
 ```
 
 Contents: [Plot an equation](#plot-an-equation) ·
@@ -224,18 +218,27 @@ ggplot2's names and defaults.
 
 ```python
 rng = np.random.default_rng(1)
-trial = pd.DataFrame({"arm": rng.choice(["placebo", "low", "high"], 150),
+arms = ["placebo", "low", "high"]
+trial = pd.DataFrame({"arm": pd.Categorical(rng.choice(arms, 150), categories=arms),
                       "dose": rng.uniform(0, 10, 150)})
-trial["response"] = 2 + 0.6 * trial.dose + 1.5 * (trial.arm == "high") + rng.normal(0, 1.2, 150)
+slope = trial.arm.map({"placebo": 0.05, "low": 0.35, "high": 0.7}).astype(float)
+trial["response"] = 2 + slope * trial.dose + rng.normal(0, 0.9, 150)
 
 a = (ggplot(trial, aes(x="dose", y="response", colour="arm"))
-     + geom_point(alpha=0.7) + geom_smooth(method="lm"))
+     + geom_point(alpha=0.7) + geom_smooth(method="lm", se=False)
+     + scale_colour_hue() + theme_grey() + theme(legend_position="bottom"))
 b = (ggplot(trial, aes(x="arm", y="response", fill="arm"))
-     + geom_violin() + geom_jitter(width=0.1, height=0) + theme(legend_position="none"))
-fig = (a | b) + plot_annotation(title="Response by dose and arm", tag_levels="A")
+     + geom_violin() + geom_jitter(width=0.1, height=0)
+     + scale_fill_hue() + theme_grey() + theme(legend_position="none"))
+fig = ((a | b) + plot_layout(widths=[3, 2])
+       + plot_annotation(title="Response by dose and arm", tag_levels="A"))
 ```
 
-<img src="https://raw.githubusercontent.com/rleyvasal/plot3/main/docs/img/ggplot2_compose.png" alt="Two-panel figure: scatter with linear fits, and violins by arm">
+`scale_colour_hue()` and `theme_grey()` are ggplot2's own defaults, so the
+figure looks as it would from R. plot3's defaults are `theme_bw()` for saved
+files and `theme_dark()` in the viewer.
+
+<img src="https://raw.githubusercontent.com/rleyvasal/plot3/main/docs/img/ggplot2_look.png" alt="Two-panel figure in ggplot2's grey theme: response against dose with one fitted line per arm, and violins of response by arm">
 
 Everything else you'd expect is there, including histograms, densities,
 boxplots, bar positions, `stat_summary`, error bars, heatmaps, hexbins, 2D
@@ -270,27 +273,35 @@ In notebooks, bare column names work as in R: `aes(x=dose, y=response)`.
 ## Install
 
 ```bash
-pip install "plot3[jupyter,export]"
+pip install plot3
 ```
 
 Python 3.10 or newer; NumPy and pandas are the only required packages.
+Notebooks, pandas, and Polars tables work with no extras.
+
+For PDF, and PNG with real fonts, add cairosvg and the Cairo library. SVG
+needs neither.
+
+```bash
+pip install cairosvg
+```
+
+```bash
+brew install cairo
+```
 
-| Extra | Adds |
-|---|---|
-| `export` | `cairosvg`: PNG with real fonts and PDF. Needs the Cairo C library (`brew install cairo`, `apt install libcairo2`). SVG needs nothing. |
-| `fast` | `contourpy`: faster implicit-curve contours (matplotlib users have it) |
-| `jupyter` | IPython integration (bare column names, `%plot3`) |
-| `polars` | Polars tables |
+(`apt install libcairo2` on Debian and Ubuntu.) `pip install "plot3[pdf]"`
+installs plot3 and cairosvg together.
 
 The latest unreleased code installs from GitHub with
-`pip install "plot3[jupyter,export] @ git+https://github.com/rleyvasal/plot3"`.
+`pip install "plot3 @ git+https://github.com/rleyvasal/plot3"`.
 
 ## Development
 
 ```bash
 git clone https://github.com/rleyvasal/plot3 && cd plot3
 python3 -m venv .venv && source .venv/bin/activate
-pip install -e ".[dev,jupyter,export]"
+pip install -e ".[dev]"
 pytest -q                       # includes every example in this README and docs/reference.md
 python examples/showcase_2d.py  # 2D gallery in the browser
 python examples/showcase_3d.py  # 3D gallery

{plot3-0.6.1 → plot3-0.6.3}/README.md

@@ -24,7 +24,7 @@ in the notebook and save as journal-ready PNG, SVG, or PDF.
 </table>
 
 ```bash
-pip install "plot3[jupyter,export]"
+pip install plot3
 ```
 
 Contents: [Plot an equation](#plot-an-equation) ·
@@ -191,18 +191,27 @@ ggplot2's names and defaults.
 
 ```python
 rng = np.random.default_rng(1)
-trial = pd.DataFrame({"arm": rng.choice(["placebo", "low", "high"], 150),
+arms = ["placebo", "low", "high"]
+trial = pd.DataFrame({"arm": pd.Categorical(rng.choice(arms, 150), categories=arms),
                       "dose": rng.uniform(0, 10, 150)})
-trial["response"] = 2 + 0.6 * trial.dose + 1.5 * (trial.arm == "high") + rng.normal(0, 1.2, 150)
+slope = trial.arm.map({"placebo": 0.05, "low": 0.35, "high": 0.7}).astype(float)
+trial["response"] = 2 + slope * trial.dose + rng.normal(0, 0.9, 150)
 
 a = (ggplot(trial, aes(x="dose", y="response", colour="arm"))
-     + geom_point(alpha=0.7) + geom_smooth(method="lm"))
+     + geom_point(alpha=0.7) + geom_smooth(method="lm", se=False)
+     + scale_colour_hue() + theme_grey() + theme(legend_position="bottom"))
 b = (ggplot(trial, aes(x="arm", y="response", fill="arm"))
-     + geom_violin() + geom_jitter(width=0.1, height=0) + theme(legend_position="none"))
-fig = (a | b) + plot_annotation(title="Response by dose and arm", tag_levels="A")
+     + geom_violin() + geom_jitter(width=0.1, height=0)
+     + scale_fill_hue() + theme_grey() + theme(legend_position="none"))
+fig = ((a | b) + plot_layout(widths=[3, 2])
+       + plot_annotation(title="Response by dose and arm", tag_levels="A"))
 ```
 
-<img src="https://raw.githubusercontent.com/rleyvasal/plot3/main/docs/img/ggplot2_compose.png" alt="Two-panel figure: scatter with linear fits, and violins by arm">
+`scale_colour_hue()` and `theme_grey()` are ggplot2's own defaults, so the
+figure looks as it would from R. plot3's defaults are `theme_bw()` for saved
+files and `theme_dark()` in the viewer.
+
+<img src="https://raw.githubusercontent.com/rleyvasal/plot3/main/docs/img/ggplot2_look.png" alt="Two-panel figure in ggplot2's grey theme: response against dose with one fitted line per arm, and violins of response by arm">
 
 Everything else you'd expect is there, including histograms, densities,
 boxplots, bar positions, `stat_summary`, error bars, heatmaps, hexbins, 2D
@@ -237,27 +246,35 @@ In notebooks, bare column names work as in R: `aes(x=dose, y=response)`.
 ## Install
 
 ```bash
-pip install "plot3[jupyter,export]"
+pip install plot3
 ```
 
 Python 3.10 or newer; NumPy and pandas are the only required packages.
+Notebooks, pandas, and Polars tables work with no extras.
+
+For PDF, and PNG with real fonts, add cairosvg and the Cairo library. SVG
+needs neither.
+
+```bash
+pip install cairosvg
+```
+
+```bash
+brew install cairo
+```
 
-| Extra | Adds |
-|---|---|
-| `export` | `cairosvg`: PNG with real fonts and PDF. Needs the Cairo C library (`brew install cairo`, `apt install libcairo2`). SVG needs nothing. |
-| `fast` | `contourpy`: faster implicit-curve contours (matplotlib users have it) |
-| `jupyter` | IPython integration (bare column names, `%plot3`) |
-| `polars` | Polars tables |
+(`apt install libcairo2` on Debian and Ubuntu.) `pip install "plot3[pdf]"`
+installs plot3 and cairosvg together.
 
 The latest unreleased code installs from GitHub with
-`pip install "plot3[jupyter,export] @ git+https://github.com/rleyvasal/plot3"`.
+`pip install "plot3 @ git+https://github.com/rleyvasal/plot3"`.
 
 ## Development
 
 ```bash
 git clone https://github.com/rleyvasal/plot3 && cd plot3
 python3 -m venv .venv && source .venv/bin/activate
-pip install -e ".[dev,jupyter,export]"
+pip install -e ".[dev]"
 pytest -q                       # includes every example in this README and docs/reference.md
 python examples/showcase_2d.py  # 2D gallery in the browser
 python examples/showcase_3d.py  # 3D gallery

plot3-0.6.3/plot3/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.6.3"

{plot3-0.6.1 → plot3-0.6.3}/plot3/build.py

@@ -762,7 +762,7 @@ def expand_stat_geom(
                 if len(group_cols) > 1:
                     out_row[colour_col] = key_tuple[1]
                 outlier_rows.append(out_row)
-        frame = pd.DataFrame(rows)
+        frame = _keep_categories(pd.DataFrame(rows), data, [colour_col])
         box_width = float(getattr(geom, "width", 0.75))
         dodge_levels = None
         if len(group_cols) > 1 and not frame.empty:
@@ -920,7 +920,7 @@ def expand_stat_geom(
                 elif colour_col == xcol:
                     row[colour_col] = x_key
                 rows.append(row)
-        frame = pd.DataFrame(rows)
+        frame = _keep_categories(pd.DataFrame(rows), data, [colour_col])
         if frame.empty:
             frame = pd.DataFrame(columns=["x", "y", "group"])
         # Force categorical x scale labels via a helper frame for domains:
@@ -1423,6 +1423,48 @@ def _legend_position_spec(value):
 
 
 
+def _categorical_order(g, resolved, levels) -> list:
+    """Colour levels in the order of a categorical colour column, as ggplot2.
+
+    Stats (density, histogram, smooth, summaries) rebuild their data, so
+    their levels arrive sorted as text. When the plot's colour column is a
+    ``pd.Categorical``, its categories set the order (and so the colours);
+    levels it does not list keep their place after them.
+    """
+    data = getattr(g, "data", None)
+    if data is None:
+        return list(levels)
+    columns = dict.fromkeys(m.get("color") for _geom, m in resolved if isinstance(m.get("color"), str))
+    for column in columns:
+        if not has_column(data, column):
+            continue
+        source = materialize_columns(data, [column])[column]
+        if not isinstance(source.dtype, pd.CategoricalDtype):
+            continue
+        wanted = [str(c) for c in source.cat.categories]
+        present = [str(level) for level in levels]
+        first = [c for c in wanted if c in present]
+        return first + [level for level in present if level not in first]
+    return list(levels)
+
+
+def _keep_categories(frame: pd.DataFrame, data, columns) -> pd.DataFrame:
+    """Give a stat's output columns the category order of the source column.
+
+    Box and violin stats build their output from plain rows, which turns
+    ``pd.Categorical(..., categories=["placebo", "low", "high"])`` into
+    text, sorted alphabetically: the fill colours then disagree with the
+    points of the same column in another panel or layer.
+    """
+    for column in dict.fromkeys(c for c in columns if c):
+        if column not in frame.columns or not has_column(data, column):
+            continue
+        source = materialize_columns(data, [column])[column]
+        if isinstance(source.dtype, pd.CategoricalDtype):
+            frame[column] = pd.Categorical(frame[column], categories=list(source.cat.categories))
+    return frame
+
+
 def _forced_levels(g, ccats) -> list:
     """Colour levels shared by every facet panel, so a group keeps its colour."""
     forced = getattr(g, "_force_color_levels", None)
@@ -2826,6 +2868,7 @@ def build_spec(g: ggplot) -> tuple[dict, list[tuple[str, str]]]:
     # viridis_d scale, else the theme's colours extended past eight groups.
     cat_colours: list[str] = []
     if color_scale is not None and color_scale[0] == "cat":
+        color_scale[1] = _categorical_order(g, resolved, color_scale[1])
         user_scale = getattr(g, "colour_scale", None)
         if user_scale is not None and user_scale.kind == "continuous":
             raise ValueError(

{plot3-0.6.1 → plot3-0.6.3}/plot3/contour.py

@@ -115,7 +115,7 @@ def _backend_name() -> str:
         if _load_contourpy() is None:
             raise ValueError(
                 "PLOT3_CONTOUR_BACKEND=contourpy but contourpy>=1.0 is not "
-                "installed. Install it with: pip install plot3[fast]"
+                "installed. Install it with: pip install contourpy"
             )
         return "contourpy"
     if _load_contourpy() is None:
@@ -346,7 +346,7 @@ def _contour_contourpy(
     if contourpy is None:
         raise ValueError(
             "PLOT3_CONTOUR_BACKEND=contourpy but contourpy>=1.0 is not "
-            "installed. Install it with: pip install plot3[fast]"
+            "installed. Install it with: pip install contourpy"
         )
     masked = np.ma.masked_invalid(np.asarray(field, dtype=np.float64))
     generator = contourpy.contour_generator(

{plot3-0.6.1 → plot3-0.6.3}/plot3/ggplot.py

@@ -517,7 +517,7 @@ class ggplot:
         suffix writes the standalone HTML viewer. ``width`` and ``height``
         are pixels unless ``units`` is ``"in"``, ``"cm"``, or ``"mm"``.
         ``dpi`` is the PNG resolution for a physical size (default 300).
-        PDF needs ``plot3[export]`` and the Cairo C library. ``.svg``
+        PDF needs cairosvg and the Cairo C library. ``.svg``
         needs neither.
         """
         suffix = Path(path).suffix.lower()
@@ -683,8 +683,8 @@ def ggsave(
     """Save ``plot``.
 
     ``ggsave("fig.png", p)`` and ``ggsave("fig.svg", p)`` write a static
-    image. ``ggsave("fig.pdf", p)`` writes a PDF when ``plot3[export]`` is
-    installed. That extra also needs the Cairo C library; ``.svg`` writes
+    image. ``ggsave("fig.pdf", p)`` writes a PDF when cairosvg is
+    installed (``plot3[pdf]``). It also needs the Cairo C library; ``.svg`` writes
     the same drawing with no extra dependencies. Any other suffix,
     including ``.html``, writes the standalone viewer. The plot and the
     filename can be passed in either order.

{plot3-0.6.1 → plot3-0.6.3}/plot3/static.py

@@ -2,7 +2,7 @@
 
 PNG, SVG, and PDF replay one list of drawing commands. SVG is text.
 PNG and PDF are that same SVG rendered by cairosvg when the optional
-``plot3[export]`` extra is installed, so the three files share one
+``plot3[pdf]`` extra (cairosvg) is installed, so the three files share one
 drawing and one font. That extra needs the Cairo C library as well as
 the Python package (``libcairo2``, or the GTK runtime on Windows).
 Without cairosvg, PNG falls back to a zlib RGB file and a built-in
@@ -42,7 +42,7 @@ _UNIT_INCH = {
 }
 _EXPORT_HINT = (
     "ggsave() needs the Cairo C library to write PDF with a journal font. "
-    "Install the Python extra with: pip install 'plot3[export]'. "
+    "Install it with: pip install cairosvg. "
     "Windows and minimal Linux images also need the Cairo library itself "
     "(for example the libcairo2 package, or the GTK runtime). "
     "ggsave('fig.svg', plot) writes the same drawing with no extra dependencies."
@@ -196,8 +196,8 @@ def save_static(
     ``"cm"`` and ``"mm"``) with ``dpi`` (default 300) sets a physical
     page. Layout stays in CSS pixels (96 per inch) so type keeps its
     size, and cairosvg rasterizes that SVG at ``dpi``. PDF and a
-    journal-font PNG need the Cairo C library (``pip install
-    'plot3[export]'``). ``.svg`` needs nothing extra.
+    journal-font PNG need cairosvg and the Cairo C library (``pip install
+    cairosvg``). ``.svg`` needs nothing extra.
     """
     path = Path(path)
     suffix = path.suffix.lower()
@@ -251,7 +251,7 @@ def save_static(
     if fallback:
         print(
             "plot3: PNG used the built-in font. "
-            "pip install 'plot3[export]' for Helvetica."
+            "pip install cairosvg for Helvetica."
         )
     return str(path)
 
@@ -2196,13 +2196,15 @@ def _paint_legend(commands, origin, metrics, theme, fonts) -> None:
     surface = theme.get("surface") or "#0b1020"
     grid = theme.get("grid") or "#1c2742"
     if _rgb(grid) == _rgb(surface):
-        grid = theme.get("muted") or "#898781"
+        # theme_grey / theme_classic: white grid on a white page. ggplot2
+        # draws no legend box, so neither do we (not a dark one instead).
+        grid = None
     tick = fonts[0]
     row_h = metrics["row_h"]
     lx, ly = origin
     # Opaque, so a legend inside the panel does not fade the marks under it.
     commands.append((
-        "rect", lx, ly, metrics["w"], metrics["h"], surface, grid, 1, 1.0,
+        "rect", lx, ly, metrics["w"], metrics["h"], surface, grid, 1 if grid else 0, 1.0,
     ))
     cursor = ly + 6
     for row in metrics["rows"]:

{plot3-0.6.1 → plot3-0.6.3}/plot3.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plot3
-Version: 0.6.1
+Version: 0.6.3
 Summary: ggplot2 grammar of graphics for Python: interactive WebGL figures and journal-ready PNG, SVG, and PDF
 Author: Rigoberto Leyva Salmeron
 License-Expression: MIT
@@ -17,18 +17,12 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy>=1.24
 Requires-Dist: pandas>=2.0
-Provides-Extra: fast
-Requires-Dist: contourpy>=1.0; extra == "fast"
+Provides-Extra: pdf
+Requires-Dist: cairosvg>=2.7; extra == "pdf"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0; extra == "dev"
 Requires-Dist: polars>=1.0; extra == "dev"
 Requires-Dist: contourpy>=1.0; extra == "dev"
-Provides-Extra: jupyter
-Requires-Dist: ipython>=8.0; extra == "jupyter"
-Provides-Extra: polars
-Requires-Dist: polars>=1.0; extra == "polars"
-Provides-Extra: export
-Requires-Dist: cairosvg>=2.7; extra == "export"
 Dynamic: license-file
 
 # plot3
@@ -57,7 +51,7 @@ in the notebook and save as journal-ready PNG, SVG, or PDF.
 </table>
 
 ```bash
-pip install "plot3[jupyter,export]"
+pip install plot3
 ```
 
 Contents: [Plot an equation](#plot-an-equation) ·
@@ -224,18 +218,27 @@ ggplot2's names and defaults.
 
 ```python
 rng = np.random.default_rng(1)
-trial = pd.DataFrame({"arm": rng.choice(["placebo", "low", "high"], 150),
+arms = ["placebo", "low", "high"]
+trial = pd.DataFrame({"arm": pd.Categorical(rng.choice(arms, 150), categories=arms),
                       "dose": rng.uniform(0, 10, 150)})
-trial["response"] = 2 + 0.6 * trial.dose + 1.5 * (trial.arm == "high") + rng.normal(0, 1.2, 150)
+slope = trial.arm.map({"placebo": 0.05, "low": 0.35, "high": 0.7}).astype(float)
+trial["response"] = 2 + slope * trial.dose + rng.normal(0, 0.9, 150)
 
 a = (ggplot(trial, aes(x="dose", y="response", colour="arm"))
-     + geom_point(alpha=0.7) + geom_smooth(method="lm"))
+     + geom_point(alpha=0.7) + geom_smooth(method="lm", se=False)
+     + scale_colour_hue() + theme_grey() + theme(legend_position="bottom"))
 b = (ggplot(trial, aes(x="arm", y="response", fill="arm"))
-     + geom_violin() + geom_jitter(width=0.1, height=0) + theme(legend_position="none"))
-fig = (a | b) + plot_annotation(title="Response by dose and arm", tag_levels="A")
+     + geom_violin() + geom_jitter(width=0.1, height=0)
+     + scale_fill_hue() + theme_grey() + theme(legend_position="none"))
+fig = ((a | b) + plot_layout(widths=[3, 2])
+       + plot_annotation(title="Response by dose and arm", tag_levels="A"))
 ```
 
-<img src="https://raw.githubusercontent.com/rleyvasal/plot3/main/docs/img/ggplot2_compose.png" alt="Two-panel figure: scatter with linear fits, and violins by arm">
+`scale_colour_hue()` and `theme_grey()` are ggplot2's own defaults, so the
+figure looks as it would from R. plot3's defaults are `theme_bw()` for saved
+files and `theme_dark()` in the viewer.
+
+<img src="https://raw.githubusercontent.com/rleyvasal/plot3/main/docs/img/ggplot2_look.png" alt="Two-panel figure in ggplot2's grey theme: response against dose with one fitted line per arm, and violins of response by arm">
 
 Everything else you'd expect is there, including histograms, densities,
 boxplots, bar positions, `stat_summary`, error bars, heatmaps, hexbins, 2D
@@ -270,27 +273,35 @@ In notebooks, bare column names work as in R: `aes(x=dose, y=response)`.
 ## Install
 
 ```bash
-pip install "plot3[jupyter,export]"
+pip install plot3
 ```
 
 Python 3.10 or newer; NumPy and pandas are the only required packages.
+Notebooks, pandas, and Polars tables work with no extras.
+
+For PDF, and PNG with real fonts, add cairosvg and the Cairo library. SVG
+needs neither.
+
+```bash
+pip install cairosvg
+```
+
+```bash
+brew install cairo
+```
 
-| Extra | Adds |
-|---|---|
-| `export` | `cairosvg`: PNG with real fonts and PDF. Needs the Cairo C library (`brew install cairo`, `apt install libcairo2`). SVG needs nothing. |
-| `fast` | `contourpy`: faster implicit-curve contours (matplotlib users have it) |
-| `jupyter` | IPython integration (bare column names, `%plot3`) |
-| `polars` | Polars tables |
+(`apt install libcairo2` on Debian and Ubuntu.) `pip install "plot3[pdf]"`
+installs plot3 and cairosvg together.
 
 The latest unreleased code installs from GitHub with
-`pip install "plot3[jupyter,export] @ git+https://github.com/rleyvasal/plot3"`.
+`pip install "plot3 @ git+https://github.com/rleyvasal/plot3"`.
 
 ## Development
 
 ```bash
 git clone https://github.com/rleyvasal/plot3 && cd plot3
 python3 -m venv .venv && source .venv/bin/activate
-pip install -e ".[dev,jupyter,export]"
+pip install -e ".[dev]"
 pytest -q                       # includes every example in this README and docs/reference.md
 python examples/showcase_2d.py  # 2D gallery in the browser
 python examples/showcase_3d.py  # 3D gallery

{plot3-0.6.1 → plot3-0.6.3}/plot3.egg-info/requires.txt

@@ -6,14 +6,5 @@ pytest>=7.0
 polars>=1.0
 contourpy>=1.0
 
-[export]
+[pdf]
 cairosvg>=2.7
-
-[fast]
-contourpy>=1.0
-
-[jupyter]
-ipython>=8.0
-
-[polars]
-polars>=1.0

{plot3-0.6.1 → plot3-0.6.3}/pyproject.toml

@@ -25,11 +25,8 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-fast = ["contourpy>=1.0"]
+pdf = ["cairosvg>=2.7"]
 dev = ["pytest>=7.0", "polars>=1.0", "contourpy>=1.0"]
-jupyter = ["ipython>=8.0"]
-polars = ["polars>=1.0"]
-export = ["cairosvg>=2.7"]
 
 [project.urls]
 Homepage = "https://github.com/rleyvasal/plot3"

{plot3-0.6.1 → plot3-0.6.3}/tests/test_contour.py

@@ -265,7 +265,7 @@ def test_version_floor_and_backend_selection(monkeypatch):
     assert _backend_name() == "numpy"
     monkeypatch.setenv("PLOT3_CONTOUR_BACKEND", "contourpy")
     _reset_contour_backend()
-    with pytest.raises(ValueError, match=r"plot3\[fast\]"):
+    with pytest.raises(ValueError, match=r"pip install contourpy"):
         _contour_lines(np.linspace(-1, 1, 4), np.linspace(-1, 1, 4), np.zeros((4, 4)), 0.0)
 
     fake = types.ModuleType("contourpy")

{plot3-0.6.1 → plot3-0.6.3}/tests/test_grammar.py

@@ -135,3 +135,45 @@ def test_geom_smooth_takes_a_linetype():
     assert [layer.get("dash") for layer in spec["layers"] if layer["kind"] == "line"] == [[4.0, 4.0]]
     with pytest.raises(ValueError):
         geom_smooth(linetype="wavy")
+
+
+@pytest.mark.parametrize("kind", ["violin", "boxplot", "density", "histogram", "bar", "smooth", "summary"])
+def test_categorical_colour_order_survives_stats(kind):
+    # pd.Categorical(categories=["placebo", "low", "high"]) sets the colour
+    # order for every geom, as in ggplot2; stats used to sort it as text, so
+    # a violin and a scatter of the same column disagreed on colours.
+    import numpy as np
+    import pandas as pd
+
+    from plot3 import (aes, geom_bar, geom_boxplot, geom_density, geom_histogram,
+                       geom_smooth, geom_violin, ggplot, stat_summary)
+    from plot3.build import build_spec
+
+    rng = np.random.default_rng(0)
+    arms = ["placebo", "low", "high"]
+    df = pd.DataFrame({"arm": pd.Categorical(rng.choice(arms, 90), categories=arms),
+                       "x": rng.uniform(0, 10, 90), "y": rng.normal(size=90),
+                       "sex": rng.choice(["F", "M"], 90)})
+    layer = {
+        "violin": (aes(x="arm", y="y", fill="arm"), geom_violin()),
+        "boxplot": (aes(x="sex", y="y", fill="arm"), geom_boxplot()),
+        "density": (aes(x="y", colour="arm"), geom_density()),
+        "histogram": (aes(x="y", fill="arm"), geom_histogram(bins=10)),
+        "bar": (aes(x="sex", fill="arm"), geom_bar()),
+        "smooth": (aes(x="x", y="y", colour="arm"), geom_smooth(method="lm")),
+        "summary": (aes(x="sex", y="y", colour="arm"), stat_summary()),
+    }[kind]
+    spec, _ = build_spec(ggplot(df, layer[0]) + layer[1])
+    assert spec["color"]["cats"] == arms
+    assert [entry["label"] for entry in spec["legend"]] == arms
+
+
+def test_theme_grey_legend_has_no_dark_frame(tmp_path):
+    import pandas as pd
+
+    from plot3 import aes, geom_point, ggplot, ggsave, theme_grey
+
+    df = pd.DataFrame({"x": [1, 2], "y": [1, 2], "g": ["a", "b"]})
+    out = tmp_path / "g.svg"
+    ggsave(str(out), ggplot(df, aes(x="x", y="y", colour="g")) + geom_point() + theme_grey())
+    assert 'stroke="#4d4d4d"' not in out.read_text()

{plot3-0.6.1 → plot3-0.6.3}/tests/test_static_save.py

@@ -438,7 +438,7 @@ def test_inches_and_dpi_set_the_png_page(tmp_path, monkeypatch, capsys):
     rgb = _read_png(path)
     assert rgb.shape == (1200, 2100, 3)
     assert tuple(rgb[0, 0]) == (255, 255, 255)
-    assert "plot3[export]" in capsys.readouterr().out
+    assert "pip install cairosvg" in capsys.readouterr().out
 
     cm = tmp_path / "cm.png"
     ggsave(cm, fig, width=2.54, height=2.54, units="cm", dpi=100)
@@ -466,7 +466,7 @@ def test_pdf_without_cairosvg_names_the_extra(tmp_path, monkeypatch):
     from plot3 import static
 
     monkeypatch.setattr(static, "_load_cairosvg", lambda: None)
-    with pytest.raises(RuntimeError, match=r"plot3\[export\]") as caught:
+    with pytest.raises(RuntimeError, match=r"pip install cairosvg") as caught:
         ggsave(tmp_path / "fig.pdf", _curve())
     message = str(caught.value)
     assert "Cairo" in message

plot3-0.6.1/plot3/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.6.1"