Skip to content

Commit 5c5dded

Browse files
committed
fix value_count test
1 parent ddcb334 commit 5c5dded

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

pandas/tests/extension/base/methods.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -30,12 +30,11 @@ def test_value_counts(self, all_data, dropna):
3030

3131
def test_value_counts_with_normalize(self, data):
3232
# GH 33172
33-
data = data[:10].unique()
33+
data = data[:10].unique().remove_unused_categories()
3434
values = np.array(data[~data.isna()])
3535
ser = pd.Series(data, dtype=data.dtype)
3636

3737
result = ser.value_counts(normalize=True).sort_index()
38-
result = result[result > 0]
3938

4039
expected = pd.Series([1 / len(values)] * len(values), index=result.index)
4140
self.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)