예제 #1
0
 def test_bool_sum_min_count(self):
     spar_bool = SparseArray([False, True] * 5,
                             dtype=np.bool8,
                             fill_value=True)
     res = spar_bool.sum(min_count=1)
     assert res == 5
     res = spar_bool.sum(min_count=11)
     assert isna(res)
예제 #2
0
 def test_sum_min_count(self, arr, fill_value, min_count, expected):
     # https://github.com/pandas-dev/pandas/issues/25777
     sparray = SparseArray(arr, fill_value=fill_value)
     result = sparray.sum(min_count=min_count)
     if np.isnan(expected):
         assert np.isnan(result)
     else:
         assert result == expected
예제 #3
0
 def test_sum_min_count(self, arr, fill_value, min_count, expected):
     # GH#25777
     sparray = SparseArray(arr, fill_value=fill_value)
     result = sparray.sum(min_count=min_count)
     if np.isnan(expected):
         assert np.isnan(result)
     else:
         assert result == expected