13
13
14
14
import hypothesis .extra .numpy as npst
15
15
import hypothesis .strategies as st
16
- from hypothesis import HealthCheck , assume , given , settings
16
+ from hypothesis import assume , given , settings
17
17
18
18
from zarr .abc .store import Store
19
19
from zarr .core .common import ZARR_JSON , ZARRAY_JSON , ZATTRS_JSON
@@ -76,7 +76,6 @@ def deep_equal(a: Any, b: Any) -> bool:
76
76
return a == b
77
77
78
78
79
- @settings (deadline = None ) # Increased from default 200ms to None
80
79
@given (data = st .data (), zarr_format = zarr_formats )
81
80
def test_array_roundtrip (data : st .DataObject , zarr_format : int ) -> None :
82
81
nparray = data .draw (numpy_arrays (zarr_formats = st .just (zarr_format )))
@@ -118,11 +117,10 @@ def test_basic_indexing(data: st.DataObject) -> None:
118
117
assert_array_equal (nparray , zarray [:])
119
118
120
119
121
- @settings (deadline = None , suppress_health_check = [HealthCheck .too_slow ])
122
120
@given (data = st .data ())
123
121
def test_oindex (data : st .DataObject ) -> None :
124
122
# integer_array_indices can't handle 0-size dimensions.
125
- zarray = data .draw (simple_arrays (shapes = npst .array_shapes (max_dims = 3 , min_side = 1 , max_side = 8 )))
123
+ zarray = data .draw (simple_arrays (shapes = npst .array_shapes (max_dims = 4 , min_side = 1 )))
126
124
nparray = zarray [:]
127
125
128
126
zindexer , npindexer = data .draw (orthogonal_indices (shape = nparray .shape ))
@@ -140,16 +138,15 @@ def test_oindex(data: st.DataObject) -> None:
140
138
assert_array_equal (nparray , zarray [:])
141
139
142
140
143
- @settings (deadline = None , suppress_health_check = [HealthCheck .too_slow ])
144
141
@given (data = st .data ())
145
142
def test_vindex (data : st .DataObject ) -> None :
146
143
# integer_array_indices can't handle 0-size dimensions.
147
- zarray = data .draw (simple_arrays (shapes = npst .array_shapes (max_dims = 3 , min_side = 1 , max_side = 8 )))
144
+ zarray = data .draw (simple_arrays (shapes = npst .array_shapes (max_dims = 4 , min_side = 1 )))
148
145
nparray = zarray [:]
149
146
150
147
indexer = data .draw (
151
148
npst .integer_array_indices (
152
- shape = nparray .shape , result_shape = npst .array_shapes (min_side = 1 , max_dims = 2 , max_side = 8 )
149
+ shape = nparray .shape , result_shape = npst .array_shapes (min_side = 1 , max_dims = None )
153
150
)
154
151
)
155
152
actual = zarray .vindex [indexer ]
0 commit comments