Skip to content

Commit dc6a26c

Browse files
committed
test: fix test for old numpy version and add finite value test
1 parent b9923db commit dc6a26c

File tree

1 file changed

+16
-8
lines changed

1 file changed

+16
-8
lines changed

pandas/tests/groupby/test_libgroupby.py

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -337,20 +337,28 @@ def test_cython_group_sum_Inf_at_beginning_and_end(values, out):
337337
(np.finfo(np.float64).max, [[np.inf]]),
338338
(np.finfo(np.float64).min, [[-np.inf]]),
339339
(
340-
np.complex128(np.finfo(np.float64).min, np.finfo(np.float64).max),
341-
[[np.complex128(-np.inf, np.inf)]],
340+
np.complex128(np.finfo(np.float64).min + np.finfo(np.float64).max * 1j),
341+
[[complex(-np.inf, np.inf)]],
342342
),
343343
(
344-
np.complex128(np.finfo(np.float64).max, np.finfo(np.float64).min),
345-
[[np.complex128(np.inf, -np.inf)]],
344+
np.complex128(np.finfo(np.float64).max + np.finfo(np.float64).min * 1j),
345+
[[complex(np.inf, -np.inf)]],
346346
),
347347
(
348-
np.complex128(np.finfo(np.float64).max, np.finfo(np.float64).max),
349-
[[np.complex128(np.inf, np.inf)]],
348+
np.complex128(np.finfo(np.float64).max + np.finfo(np.float64).max * 1j),
349+
[[complex(np.inf, np.inf)]],
350350
),
351351
(
352-
np.complex128(np.finfo(np.float64).min, np.finfo(np.float64).min),
353-
[[np.complex128(-np.inf, -np.inf)]],
352+
np.complex128(np.finfo(np.float64).min + np.finfo(np.float64).min * 1j),
353+
[[complex(-np.inf, -np.inf)]],
354+
),
355+
(
356+
np.complex128(3.0 + np.finfo(np.float64).min * 1j),
357+
[[complex(9.0, -np.inf)]],
358+
),
359+
(
360+
np.complex128(np.finfo(np.float64).max + 3 * 1j),
361+
[[complex(np.inf, 9.0)]],
354362
),
355363
],
356364
)

0 commit comments

Comments
 (0)