@@ -11,11 +11,11 @@ def test_slice_3D(make_napari_viewer, brain_data):
1111 viewer .theme = "light"
1212
1313 data = brain_data [0 ]
14- assert data .ndim == 3
15- viewer .add_image (brain_data [ 0 ] , ** brain_data [1 ])
14+ assert data .ndim == 3 , data . shape
15+ viewer .add_image (data , ** brain_data [1 ])
1616
1717 axis = viewer .dims .last_used
18- slice_no = brain_data [ 0 ] .shape [0 ] - 1
18+ slice_no = data .shape [0 ] - 1
1919 viewer .dims .set_current_step (axis , slice_no )
2020 fig = SliceWidget (viewer ).figure
2121 # Need to return a copy, as original figure is too eagerley garbage
@@ -28,9 +28,10 @@ def test_slice_2D(make_napari_viewer, astronaut_data):
2828 viewer = make_napari_viewer ()
2929 viewer .theme = "light"
3030
31- data = astronaut_data [0 ]
32- assert data .ndim == 2
33- viewer .add_image (astronaut_data [0 ], ** astronaut_data [1 ])
31+ # Take first RGB channel
32+ data = astronaut_data [0 ][:, :, 0 ]
33+ assert data .ndim == 2 , data .shape
34+ viewer .add_image (data , ** astronaut_data [1 ])
3435
3536 fig = SliceWidget (viewer ).figure
3637 # Need to return a copy, as original figure is too eagerley garbage
0 commit comments