Merge pull request #152 from ruaridhg/scatter-test-3D-data

3D scatter test

Author: ruaridhg

Diff for: src/napari_matplotlib/tests/baseline/test_histogram_3D.png

@@ -23,3 +23,26 @@ 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()
+    axis = viewer.dims.last_used
+    slice_no = 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)

Diff for: src/napari_matplotlib/tests/baseline/test_slice_3D.png

@@ -23,6 +23,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 = 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

Diff for: src/napari_matplotlib/tests/scatter/baseline/test_scatter_3D.png

@@ -9,6 +9,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 = 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