@@ -125,7 +125,7 @@ def test_max(x, data):
125
125
s = scalar_type (x [idx ])
126
126
elements .append (s )
127
127
expected = max (elements )
128
- assert_equals ("max" , dh . get_scalar_type ( out . dtype ) , out_idx , max_ , expected )
128
+ assert_equals ("max" , scalar_type , out_idx , max_ , expected )
129
129
130
130
131
131
@given (
@@ -154,7 +154,7 @@ def test_mean(x, data):
154
154
s = float (x [idx ])
155
155
elements .append (s )
156
156
expected = sum (elements ) / len (elements )
157
- assert_equals ("mean" , dh . get_scalar_type ( out . dtype ) , out_idx , mean , expected )
157
+ assert_equals ("mean" , float , out_idx , mean , expected )
158
158
159
159
160
160
@given (
@@ -183,7 +183,7 @@ def test_min(x, data):
183
183
s = scalar_type (x [idx ])
184
184
elements .append (s )
185
185
expected = min (elements )
186
- assert_equals ("min" , dh . get_scalar_type ( out . dtype ) , out_idx , min_ , expected )
186
+ assert_equals ("min" , scalar_type , out_idx , min_ , expected )
187
187
188
188
189
189
@given (
@@ -246,7 +246,7 @@ def test_prod(x, data):
246
246
if dh .is_int_dtype (out .dtype ):
247
247
m , M = dh .dtype_ranges [out .dtype ]
248
248
assume (m <= expected <= M )
249
- assert_equals ("prod" , dh . get_scalar_type ( out . dtype ) , out_idx , prod , expected )
249
+ assert_equals ("prod" , scalar_type , out_idx , prod , expected )
250
250
251
251
252
252
@given (
@@ -344,7 +344,7 @@ def test_sum(x, data):
344
344
if dh .is_int_dtype (out .dtype ):
345
345
m , M = dh .dtype_ranges [out .dtype ]
346
346
assume (m <= expected <= M )
347
- assert_equals ("sum" , dh . get_scalar_type ( out . dtype ) , out_idx , sum_ , expected )
347
+ assert_equals ("sum" , scalar_type , out_idx , sum_ , expected )
348
348
349
349
350
350
@given (
0 commit comments