Paired nd arrays for test_equal()

Author: honno

array_api_tests/test_creation_functions.py

@@ -130,7 +130,6 @@ def test_full(shape, fill_value, dtype):
     else:
         assert all(equal(a, asarray(fill_value, **kwargs))), "full() array did not equal the fill value"
 
-# TODO: implement full_like (requires hypothesis arrays support)
 @given(
     a=xps.arrays(
         dtype=shared(xps.scalar_dtypes(), key='dtypes'),

array_api_tests/test_elementwise_functions.py

@@ -15,7 +15,7 @@
 """
 
 from hypothesis import given, assume
-from hypothesis.strategies import composite, just
+from hypothesis.strategies import composite, just, shared
 
 import math
 
@@ -26,7 +26,7 @@
                                  boolean_dtype_objects, floating_dtypes,
                                  numeric_dtypes, integer_or_boolean_dtypes,
                                  boolean_dtypes, mutually_promotable_dtypes,
-                                 array_scalars)
+                                 array_scalars, xps)
 from .array_helpers import (assert_exactly_equal, negative,
                             positive_mathematical_sign,
                             negative_mathematical_sign, logical_not,
@@ -361,9 +361,17 @@ def test_divide(args):
     # could test that this does implement IEEE 754 division, but we don't yet
     # have those sorts in general for this module.
 
-@given(two_any_dtypes.flatmap(lambda i: two_array_scalars(*i)))
-def test_equal(args):
-    x1, x2 = args
+
+
+@given(
+    x1=shared(
+        xps.arrays(dtype=xps.scalar_dtypes(), shape=xps.array_shapes()), key='arrays'
+    ),
+    x2=shared(
+        xps.arrays(dtype=xps.scalar_dtypes(), shape=xps.array_shapes()), key='arrays'
+    ),
+)
+def test_equal(x1, x2):
     sanity_check(x1, x2)
     a = _array_module.equal(x1, x2)
     # NOTE: assert_exactly_equal() itself uses equal(), so we must be careful