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)
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
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