Enabled the equal_nan keyword argument for np.array_equal() when the arguments are astropy.units.Quantity instances.

astropy/units/quantity_helper/function_helpers.py

@@ -562,9 +562,9 @@ def close(a, b, rtol=1e-05, atol=1e-08, *args, **kwargs):
 
 
 @function_helper
-def array_equal(a1, a2):
+def array_equal(a1, a2, equal_nan=False):
     args, unit = _quantities2arrays(a1, a2)
-    return args, {}, None, None
+    return args, dict(equal_nan=equal_nan), None, None
 
 
 @function_helper

astropy/units/tests/test_quantity_non_ufuncs.py

@@ -924,6 +924,14 @@ def test_array_equal(self):
         q3 = q1.value * u.cm
         assert not np.array_equal(q1, q3)
 
+    @pytest.mark.parametrize("equal_nan", [False, True])
+    def test_array_equal_nan(self, equal_nan):
+        q1 = np.linspace(0, 1, num=11) * u.m
+        q1[0] = np.nan
+        q2 = q1.to(u.cm)
+        result = np.array_equal(q1, q2, equal_nan=equal_nan)
+        assert result == equal_nan
+
     @needs_array_function
     def test_array_equiv(self):
         q1 = np.array([[0.0, 1.0, 2.0]] * 3) * u.m

docs/changes/units/14135.feature.rst

@@ -0,0 +1,2 @@
+Enabled the ``equal_nan`` keyword argument for ``np.array_equal()`` when the
+arguments are ``astropy.units.Quantity`` instances.