Test multiple data shapes

Author: dstansby

Diff for: src/napari_matplotlib/tests/conftest.py

@@ -0,0 +1,13 @@
+import numpy as np
+import pytest
+
+
+@pytest.fixture(
+    params=[
+        ((100, 100), {"rgb": False}),
+        ((100, 100, 100), {"rgb": False}),
+        ((100, 100, 3), {"rgb": True}),
+    ]
+)
+def image_data(request):
+    return np.ones(request.param[0]), request.param[1]

Diff for: src/napari_matplotlib/tests/test_histogram.py

@@ -1,15 +1,8 @@
-import numpy as np
-import pytest
-
 from napari_matplotlib import HistogramWidget
 
 
-# Test with 2D and 3D data
-@pytest.mark.parametrize(
-    "data", [np.random.random((100, 100)), np.random.random((100, 100, 100))]
-)
-def test_example_q_widget(make_napari_viewer, data):
+def test_example_q_widget(make_napari_viewer, image_data):
     # Smoke test adding a histogram widget
     viewer = make_napari_viewer()
-    viewer.add_image(data)
+    viewer.add_image(image_data[0], **image_data[1])
     HistogramWidget(viewer)