3D scatter test

src/napari_matplotlib/tests/scatter/test_scatter.py

@@ -1,6 +1,7 @@
 from copy import deepcopy
 
 import pytest
+from numpy import array
 
 from napari_matplotlib import ScatterWidget
 
@@ -23,3 +24,27 @@ def test_scatter_2D(make_napari_viewer, astronaut_data):
     viewer.layers.selection.add(viewer.layers[0])
     viewer.layers.selection.add(viewer.layers[1])
     return deepcopy(fig)
+
+
+@pytest.mark.mpl_image_compare
+def test_scatter_3D(make_napari_viewer, brain_data):
+    viewer = make_napari_viewer()
+    widget = ScatterWidget(viewer)
+    fig = widget.figure
+
+    viewer.add_image(brain_data[0], **brain_data[1], name="brain")
+
+    viewer.add_image(
+        brain_data[0] * -1, **brain_data[1], name="brain_reversed"
+    )
+    # De-select existing selection
+    viewer.layers.selection.clear()
+    # pdb.set_trace()
+    axis = viewer.dims.last_used
+    slice_no = array(brain_data[0]).shape[0] - 1
+    viewer.dims.set_current_step(axis, slice_no)
+    # Select images
+    viewer.layers.selection.add(viewer.layers[0])
+    viewer.layers.selection.add(viewer.layers[1])
+
+    return deepcopy(fig)

src/napari_matplotlib/tests/test_histogram.py

@@ -1,6 +1,7 @@
 from copy import deepcopy
 
 import pytest
+from numpy import array
 
 from napari_matplotlib import HistogramWidget
 from napari_matplotlib.tests.helpers import (
@@ -23,6 +24,9 @@ def test_histogram_2D(make_napari_viewer, astronaut_data):
 def test_histogram_3D(make_napari_viewer, brain_data):
     viewer = make_napari_viewer()
     viewer.add_image(brain_data[0], **brain_data[1])
+    axis = viewer.dims.last_used
+    slice_no = array(brain_data[0]).shape[0] - 1
+    viewer.dims.set_current_step(axis, slice_no)
     fig = HistogramWidget(viewer).figure
     # Need to return a copy, as original figure is too eagerley garbage
     # collected by the widget

src/napari_matplotlib/tests/test_slice.py

@@ -1,6 +1,7 @@
 from copy import deepcopy
 
 import pytest
+from numpy import array
 
 from napari_matplotlib import SliceWidget
 
@@ -9,6 +10,9 @@
 def test_slice_3D(make_napari_viewer, brain_data):
     viewer = make_napari_viewer()
     viewer.add_image(brain_data[0], **brain_data[1])
+    axis = viewer.dims.last_used
+    slice_no = array(brain_data[0]).shape[0] - 1
+    viewer.dims.set_current_step(axis, slice_no)
     fig = SliceWidget(viewer).figure
     # Need to return a copy, as original figure is too eagerley garbage
     # collected by the widget