@@ -131,15 +131,8 @@ def test_arange(dtype, data):
131
131
size <= hh .MAX_ARRAY_SIZE
132
132
), f"{ size = } should be no more than { hh .MAX_ARRAY_SIZE } " # sanity check
133
133
134
- kw = data .draw (
135
- hh .specified_kwargs (
136
- hh .KVD ("stop" , stop , None ),
137
- hh .KVD ("step" , step , None ),
138
- hh .KVD ("dtype" , dtype , None ),
139
- ),
140
- label = "kw" ,
141
- )
142
- out = xp .arange (start , ** kw )
134
+ kw = data .draw (hh .specified_kwargs (hh .KVD ("dtype" , dtype , None )), label = "kw" )
135
+ out = xp .arange (start , stop , step , ** kw )
143
136
144
137
if dtype is None :
145
138
if all_int :
@@ -149,7 +142,10 @@ def test_arange(dtype, data):
149
142
else :
150
143
ph .assert_dtype ("arange" , (out .dtype ,), dtype )
151
144
assert out .ndim == 1 , f"{ out .ndim = } , but should be 1 [linspace()]"
152
- f_func = f"[arange({ start } , { stop } , { step } )]"
145
+ if stop is None :
146
+ f_func = f"[arange({ start } , { step = } )]"
147
+ else :
148
+ f_func = f"[arange({ start } , { stop } , { step } )]"
153
149
# We check size is roughly as expected to avoid edge cases e.g.
154
150
#
155
151
# >>> xp.arange(2, step=0.333333333333333)
0 commit comments