Skip to content

Commit d31aa79

Browse files
committed
Add test for all-NA case
1 parent a1444c9 commit d31aa79

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

pandas/tests/groupby/test_reductions.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -520,15 +520,27 @@ def test_sum_skipna_object(skipna):
520520
("prod", [0, 1, 3, np.nan, 4, 5, 6, 7, -8, 9], "float64", "float64"),
521521
("prod", [0, -1, 3, 4, 5, np.nan, 6, 7, 8, 9], "Float64", "Float64"),
522522
("prod", [0, 1, 3, -4, 5, 6, 7, -8, np.nan, 9], "Int64", "Int64"),
523+
("prod", [np.nan] * 10, "float64", "float64"),
524+
("prod", [np.nan] * 10, "Float64", "Float64"),
525+
("prod", [np.nan] * 10, "Int64", "Int64"),
523526
("var", [0, -1, 3, 4, np.nan, 5, 6, 7, 8, 9], "float64", "float64"),
524527
("var", [0, 1, 3, -4, 5, 6, 7, -8, 9, np.nan], "Float64", "Float64"),
525528
("var", [0, -1, 3, 4, 5, -6, 7, np.nan, 8, 9], "Int64", "Float64"),
529+
("var", [np.nan] * 10, "float64", "float64"),
530+
("var", [np.nan] * 10, "Float64", "Float64"),
531+
("var", [np.nan] * 10, "Int64", "Float64"),
526532
("std", [0, 1, 3, -4, 5, 6, 7, -8, np.nan, 9], "float64", "float64"),
527533
("std", [0, -1, 3, 4, 5, -6, 7, np.nan, 8, 9], "Float64", "Float64"),
528534
("std", [0, 1, 3, -4, 5, 6, 7, -8, 9, np.nan], "Int64", "Float64"),
535+
("std", [np.nan] * 10, "float64", "float64"),
536+
("std", [np.nan] * 10, "Float64", "Float64"),
537+
("std", [np.nan] * 10, "Int64", "Float64"),
529538
("sem", [0, -1, 3, 4, 5, -6, 7, np.nan, 8, 9], "float64", "float64"),
530539
("sem", [0, 1, 3, -4, 5, 6, 7, -8, np.nan, 9], "Float64", "Float64"),
531540
("sem", [0, -1, 3, 4, 5, -6, 7, 8, 9, np.nan], "Int64", "Float64"),
541+
("sem", [np.nan] * 10, "float64", "float64"),
542+
("sem", [np.nan] * 10, "Float64", "Float64"),
543+
("sem", [np.nan] * 10, "Int64", "Float64"),
532544
("min", [0, -1, 3, 4, 5, -6, 7, np.nan, 8, 9], "float64", "float64"),
533545
("min", [0, 1, 3, -4, 5, 6, 7, -8, np.nan, 9], "Float64", "Float64"),
534546
("min", [0, -1, 3, 4, 5, -6, 7, 8, 9, np.nan], "Int64", "Int64"),
@@ -557,6 +569,9 @@ def test_sum_skipna_object(skipna):
557569
"datetime64[ns]",
558570
"datetime64[ns]",
559571
),
572+
("min", [np.nan] * 10, "float64", "float64"),
573+
("min", [np.nan] * 10, "Float64", "Float64"),
574+
("min", [np.nan] * 10, "Int64", "Int64"),
560575
("max", [0, -1, 3, 4, 5, -6, 7, np.nan, 8, 9], "float64", "float64"),
561576
("max", [0, 1, 3, -4, 5, 6, 7, -8, np.nan, 9], "Float64", "Float64"),
562577
("max", [0, -1, 3, 4, 5, -6, 7, 8, 9, np.nan], "Int64", "Int64"),
@@ -585,6 +600,9 @@ def test_sum_skipna_object(skipna):
585600
"datetime64[ns]",
586601
"datetime64[ns]",
587602
),
603+
("max", [np.nan] * 10, "float64", "float64"),
604+
("max", [np.nan] * 10, "Float64", "Float64"),
605+
("max", [np.nan] * 10, "Int64", "Int64"),
588606
("median", [0, -1, 3, 4, 5, -6, 7, np.nan, 8, 9], "float64", "float64"),
589607
("median", [0, 1, 3, -4, 5, 6, 7, -8, np.nan, 9], "Float64", "Float64"),
590608
("median", [0, -1, 3, 4, 5, -6, 7, 8, 9, np.nan], "Int64", "Float64"),
@@ -613,6 +631,9 @@ def test_sum_skipna_object(skipna):
613631
"datetime64[ns]",
614632
"datetime64[ns]",
615633
),
634+
("median", [np.nan] * 10, "float64", "float64"),
635+
("median", [np.nan] * 10, "Float64", "Float64"),
636+
("median", [np.nan] * 10, "Int64", "Float64"),
616637
],
617638
)
618639
def test_multifunc_skipna(func, values, dtype, result_dtype, skipna):

0 commit comments

Comments
 (0)