Test vector plots

jonseddon · jonseddon · commit 5d67a6747042 · 2021-08-05T18:03:51.000+01:00
diff --git a/lib/iris/tests/integration/plot/test_vector_plots.py b/lib/iris/tests/integration/plot/test_vector_plots.py
@@ -24,16 +24,16 @@
 if tests.MPL_AVAILABLE:
     import matplotlib.pyplot as plt
 
-    from iris.plot import quiver
+    from iris.plot import barbs, quiver
 
 
 @tests.skip_plot
 class MixinVectorPlotCases:
     """
-    Test examples mixin, used by separate quiver + streamplot classes.
+    Test examples mixin, used by separate barb, quiver + streamplot classes.
 
-    NOTE: at present for quiver only, as streamplot does not support arbitrary
-    coordinates.
+    NOTE: at present for barb and quiver only, as streamplot does not support
+    arbitrary coordinates.
 
     """
 
@@ -193,6 +193,34 @@ def test_circular_longitude(self):
         self.plot("circular", u_cube, v_cube, coords=("longitude", "latitude"))
 
 
+class TestBarbs(MixinVectorPlotCases, tests.GraphicsTest):
+    def setUp(self):
+        super().setUp()
+
+    @staticmethod
+    def _nonlatlon_xyuv():
+        # Increase the range of wind speeds used in the barbs test to test more
+        # barbs shapes than just circles
+        x, y, u, v = MixinVectorPlotCases._nonlatlon_xyuv()
+        scale_factor = 50
+        u *= scale_factor
+        v *= scale_factor
+        return x, y, u, v
+
+    @staticmethod
+    def _latlon_uv_cubes(grid_cube):
+        # Increase the range of wind speeds used in the barbs test to test all
+        # barbs shapes
+        u_cube, v_cube = MixinVectorPlotCases._latlon_uv_cubes(grid_cube)
+        scale_factor = 30
+        u_cube.data *= scale_factor
+        v_cube.data *= scale_factor
+        return  u_cube, v_cube
+
+    def plot_function_to_test(self):
+        return barbs
+
+
 class TestQuiver(MixinVectorPlotCases, tests.GraphicsTest):
     def setUp(self):
         super().setUp()
diff --git a/lib/iris/tests/results/imagerepo.json b/lib/iris/tests/results/imagerepo.json
@@ -177,6 +177,21 @@
     "iris.tests.integration.plot.test_plot_2d_coords.Test2dContour.test_2d_coords_contour.0": [
         "https://scitools.github.io/test-iris-imagehash/images/v4/b4b2643ecb05cb43b0f23d80c53c4e1d3e5990eb1f81c19f2f983cb1c4ff3e42.png"
     ],
+    "iris.tests.integration.plot.test_vector_plots.TestBarbs.test_2d_plain_latlon.0": [
+        "https://scitools.github.io/test-iris-imagehash/images/v4/eb036726c47c9273918e6e2c6f216336787590eb969a165890ee6c676925b3b3.png"
+    ],
+    "iris.tests.integration.plot.test_vector_plots.TestBarbs.test_2d_plain_latlon_on_polar_map.0": [
+        "https://scitools.github.io/test-iris-imagehash/images/v4/e66d673c999031cd6667663398dc332c676364e798959336636660d933998666.png"
+    ],
+    "iris.tests.integration.plot.test_vector_plots.TestBarbs.test_2d_rotated_latlon.0": [
+        "https://scitools.github.io/test-iris-imagehash/images/v4/eba037a4c479c273b2963f2c6f6126966865d86f969e33c9b1706c26692793b0.png"
+    ],
+    "iris.tests.integration.plot.test_vector_plots.TestBarbs.test_non_latlon_1d_coords.0": [
+        "https://scitools.github.io/test-iris-imagehash/images/v4/a7ac334934d2e65c72596325b343338cb41c92d9c5b36f65330d379692ca6d6c.png"
+    ],
+    "iris.tests.integration.plot.test_vector_plots.TestBarbs.test_non_latlon_2d_coords.0": [
+        "https://scitools.github.io/test-iris-imagehash/images/v4/a7acb36134d2e676627963259343330cb43e92d9c5336e67330d379292ca6d6c.png"
+    ],
     "iris.tests.integration.plot.test_vector_plots.TestQuiver.test_2d_plain_latlon.0": [
         "https://scitools.github.io/test-iris-imagehash/images/v4/fb8d4f21c472b27e919d2e216f216b3178e69c7e961ab39a84696c616d245b94.png"
     ],
