matplotlib
diff --git a/‎src/napari_matplotlib/tests/test_histogram.py
Copy file name to clipboardExpand all lines: src/napari_matplotlib/tests/test_histogram.py
+38-12Lines changed: 38 additions & 12 deletions b/‎src/napari_matplotlib/tests/test_histogram.py
Copy file name to clipboardExpand all lines: src/napari_matplotlib/tests/test_histogram.py
+38-12Lines changed: 38 additions & 12 deletions
@@ -36,8 +36,11 @@ def test_histogram_3D(make_napari_viewer, brain_data):
 
 
 def test_feature_histogram(make_napari_viewer):
+
     n_points = 1000
     random_points = np.random.random((n_points, 3)) * 10
+    random_directions = np.random.random((n_points, 3)) * 10
+    random_vectors = np.stack([random_points, random_directions], axis=1)
     feature1 = np.random.random(n_points)
     feature2 = np.random.normal(size=n_points)
 
@@ -47,10 +50,10 @@ def test_feature_histogram(make_napari_viewer):
         properties={"feature1": feature1, "feature2": feature2},
         name="points1",
     )
-    viewer.add_points(
-        random_points,
+    viewer.add_vectors(
+        random_vectors,
         properties={"feature1": feature1, "feature2": feature2},
-        name="points2",
+        name="vectors1",
     )
 
     widget = FeaturesHistogramWidget(viewer)
@@ -70,26 +73,44 @@ def test_feature_histogram(make_napari_viewer):
 
 
 @pytest.mark.mpl_image_compare
-def test_feature_histogram2(make_napari_viewer):
-    import numpy as np
+def test_feature_histogram_vectors(make_napari_viewer):
+
+    n_points = 1000
+    np.random.seed(42)
+    random_points = np.random.random((n_points, 3)) * 10
+    random_directions = np.random.random((n_points, 3)) * 10
+    random_vectors = np.stack([random_points, random_directions], axis=1)
+    feature1 = np.random.random(n_points)
+
+    viewer = make_napari_viewer()
+    viewer.add_vectors(
+        random_vectors,
+        properties={"feature1": feature1},
+        name="vectors1",
+    )
+
+    widget = FeaturesHistogramWidget(viewer)
+    viewer.window.add_dock_widget(widget)
+    widget._set_axis_keys("feature1")
+
+    fig = FeaturesHistogramWidget(viewer).figure
+    return deepcopy(fig)
+
+
+@pytest.mark.mpl_image_compare
+def test_feature_histogram_points(make_napari_viewer):
 
     np.random.seed(0)
     n_points = 1000
     random_points = np.random.random((n_points, 3)) * 10
     feature1 = np.random.random(n_points)
-    feature2 = np.random.normal(size=n_points)
 
     viewer = make_napari_viewer()
     viewer.add_points(
         random_points,
-        properties={"feature1": feature1, "feature2": feature2},
+        properties={"feature1": feature1},
         name="points1",
     )
-    viewer.add_points(
-        random_points,
-        properties={"feature1": feature1, "feature2": feature2},
-        name="points2",
-    )
 
     widget = FeaturesHistogramWidget(viewer)
     viewer.window.add_dock_widget(widget)
@@ -120,3 +141,8 @@ def test_change_layer(make_napari_viewer, brain_data, astronaut_data):
     viewer.layers.selection.clear()
     viewer.layers.selection.add(viewer.layers[1])
     assert_figures_not_equal(widget.figure, fig1)
+
+
+if __name__ == '__main__':
+    import napari
+    test_feature_histogram_points(napari.Viewer)