fix #462

d-chambers · d-chambers · commit 4f661d28920b · 2025-10-04T17:32:03.000+01:00
diff --git a/dascore/viz/wiggle.py b/dascore/viz/wiggle.py
@@ -93,6 +93,22 @@ def wiggle(
     >>> _ = patch.viz.wiggle()
     """
     ax = _get_ax(ax)
+
+    # Handle 1D patches (issue #462)
+    if len(patch.dims) == 1:
+        plot_dim = patch.dims[0]
+        x_values = patch.coords.get_array(plot_dim)
+        y_values = patch.data
+        ax.plot(x_values, y_values, color=color, alpha=alpha)
+        ax.set_xlabel(_get_dim_label(patch, plot_dim))
+        ax.set_ylabel("amplitude")
+        # Format time axis if applicable
+        if np.issubdtype(patch.get_coord(plot_dim).dtype, np.datetime64):
+            _format_time_axis(ax, plot_dim, "x")
+        if show:
+            plt.show()
+        return ax
+
     assert len(patch.dims) == 2, "Can only make wiggle plot of 2D Patch"
     # After transpose selected dim must be axis 0 and other axis 1
     patch = patch.transpose(dim, ...)
diff --git a/tests/test_viz/test_wiggle.py b/tests/test_viz/test_wiggle.py
@@ -54,3 +54,19 @@ def test_show(self, random_patch, monkeypatch):
         """Ensure show path is callable."""
         monkeypatch.setattr(plt, "show", lambda: None)
         random_patch.viz.wiggle(show=True)
+
+    def test_1d_patch(self, random_patch):
+        """Test that wiggle works with 1D patches (issue #462)."""
+        # Create a 1D patch by reducing one dimension
+        patch_1d = random_patch.mean("distance", dim_reduce="squeeze")
+        # This should work without raising an assertion error
+        ax = patch_1d.viz.wiggle()
+        assert isinstance(ax, plt.Axes)
+        # The remaining dimension should be on the x-axis
+        assert patch_1d.dims[0].lower() in ax.get_xlabel().lower()
+
+    def test_1d_patch_show(self, random_patch, monkeypatch):
+        """Test that show works with 1D patches (issue #462)."""
+        monkeypatch.setattr(plt, "show", lambda: None)
+        patch_1d = random_patch.mean("distance", dim_reduce="squeeze")
+        patch_1d.viz.wiggle(show=True)
