ENH: Check that the lengths of the inputs to histogram2d are the same. (numpy#20228)

Improves exception message when inputs have different shapes.

Closes numpygh-20050

Co-authored-by: Ross Barnowski <[email protected]>

Author: jeertmans

numpy/lib/tests/test_twodim_base.py

@@ -18,6 +18,9 @@
 from numpy.core.tests.test_overrides import requires_array_function
 
 
+import pytest
+
+
 def get_mat(n):
     data = arange(n)
     data = add.outer(data, data)
@@ -295,6 +298,13 @@ def __array_function__(self, function, types, args, kwargs):
         r = histogram2d(xy, xy, weights=s_d)
         assert_(r, ((ShouldDispatch,), (xy, xy), dict(weights=s_d)))
 
+    @pytest.mark.parametrize(("x_len", "y_len"), [(10, 11), (20, 19)])
+    def test_bad_length(self, x_len, y_len):
+        x, y = np.ones(x_len), np.ones(y_len)
+        with pytest.raises(ValueError,
+                           match='x and y must have the same length.'):
+            histogram2d(x, y)
+
 
 class TestTri:
     def test_dtype(self):

numpy/lib/twodim_base.py

@@ -804,6 +804,9 @@ def histogram2d(x, y, bins=10, range=None, normed=None, weights=None,
     >>> plt.show()
     """
     from numpy import histogramdd
+    
+    if len(x) != len(y):
+        raise ValueError('x and y must have the same length.')
 
     try:
         N = len(bins)