Fixing missed changes with ak.array_equal

* Now overrides numpy.array_equal * Adding link to toctree * Unit test checks numpy override
scikit-hep · Aug 23, 2024 · f9304c9 · f9304c9
1 parent 769ee40
commit f9304c9
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/docs/reference/toctree.txt b/docs/reference/toctree.txt
@@ -299,11 +299,12 @@
     generated/ak.backend
 
 .. toctree::
-    :caption: Approximation
+    :caption: Approximation and comparison
 
     generated/ak.round
     generated/ak.isclose
     generated/ak.almost_equal
+    generated/ak.array_equal
 
 .. toctree::
     :caption: NumPy compatibility

diff --git a/src/awkward/operations/ak_array_equal.py b/src/awkward/operations/ak_array_equal.py
@@ -8,6 +8,7 @@
 __all__ = ("array_equal",)
 
 
+@ak._connect.numpy.implements("array_equal")
 @high_level_function()
 def array_equal(
     a1,

diff --git a/tests/test_1105_ak_aray_equal.py b/tests/test_1105_ak_aray_equal.py
@@ -88,3 +88,10 @@ def test_array_equal_with_params():
     )
     assert not ak.array_equal(a1, a2)
     assert ak.array_equal(a1, a2, check_parameters=False)
+
+
+def test_array_equal_numpy_override():
+    assert np.array_equal(
+        ak.Array([[1, 2], [], [3, 4, 5]]),
+        ak.Array([[1, 2], [], [3, 4, 5]]),
+    )