Skip to content

Commit 2acbbc2

Browse files
committed
Make numeric_arrays multi-dimensional
1 parent fc1e7f6 commit 2acbbc2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

array_api_tests/hypothesis_helpers.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -134,8 +134,10 @@ def two_broadcastable_shapes(draw, shapes=shapes):
134134

135135
nonbroadcastable_ones_array_two_args = hypotheses_tuples(ones_arrays, ones_arrays)
136136

137-
# TODO: Generate general arrays here, rather than just scalars.
138-
numeric_arrays = builds(full, just((1,)), floats())
137+
numeric_arrays = xps.arrays(
138+
dtype=shared(xps.floating_dtypes(), key='dtypes'),
139+
shape=shared(xps.array_shapes(), key='shapes'),
140+
)
139141

140142
@composite
141143
def scalars(draw, dtypes, finite=False):

0 commit comments

Comments
 (0)