fix for hemisphere='lower'

Author: MACarlsen

orix/measure/pole_density_function.py

@@ -106,6 +106,7 @@ def pole_density_function(
     elif len(args) == 2:
         # azimuth and polar angles
         v = Vector3d.from_polar(*args)
+
     else:
         raise ValueError(
             "Accepts only one (Vector3d) or two (azimuth, polar)\
@@ -186,7 +187,7 @@ def pole_density_function(
     # Make plot grid
     azimuth_coords, polar_coords = _sample_S2_uv_mesh_coordinates(
         plot_resolution,
-        hemisphere="upper",
+        hemisphere=hemisphere,
         azimuth_endpoint=True,
     )
     azimuth_grid, polar_grid = np.meshgrid(

orix/tests/test_measure/test_measure.py

@@ -68,6 +68,12 @@ def test_output_format(self):
         assert isinstance(hist2, np.ma.MaskedArray)
         assert hist2.mask.sum() > 0
 
+        hist3, (x3, y3) = pole_density_function(v, hemisphere="lower")
+        assert hist3.shape[0] + 1 == x3.shape[0] == y3.shape[0]
+        assert hist3.shape[1] + 1 == x3.shape[1] == y3.shape[1]
+        assert isinstance(hist3, np.ma.MaskedArray)
+        assert hist3.mask.sum() == 0
+
     @pytest.mark.parametrize("resolution", [0.5, 1.0])
     def test_pole_density_function_mrd_norm(self, point_groups, resolution):
         pg = point_groups