diff --git a/docs/reference/toctree.txt b/docs/reference/toctree.txt index 126e0a2b9c..4b6ae1154d 100644 --- 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 index 2a7221baab..398db6b2a6 100644 --- 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 index 519d0039c5..7512f50316 100644 --- 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]]), + )