Skip to content

Commit 0394389

Browse files
committed
Make numeric_arrays multi-dimensional
1 parent 5359c8f commit 0394389

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

array_api_tests/hypothesis_helpers.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,10 @@ def two_broadcastable_shapes(draw, shapes=shapes):
130130
sqrt_sizes = integers(0, SQRT_MAX_ARRAY_SIZE)
131131

132132
# TODO: Generate general arrays here, rather than just scalars.
133-
numeric_arrays = builds(full, just((1,)), floats())
133+
numeric_arrays = xps.arrays(
134+
dtype=shared(xps.floating_dtypes(), key='dtypes'),
135+
shape=shared(xps.array_shapes(), key='shapes'),
136+
)
134137

135138
@composite
136139
def scalars(draw, dtypes, finite=False):

0 commit comments

Comments
 (0)