Skip to content

Commit cba219c

Browse files
committed
fix value_count test
1 parent ddcb334 commit cba219c

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

pandas/tests/extension/base/methods.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -30,12 +30,10 @@ 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()])
35-
ser = pd.Series(data, dtype=data.dtype)
3635

37-
result = ser.value_counts(normalize=True).sort_index()
38-
result = result[result > 0]
36+
result = pd.Series(data).value_counts(normalize=True).sort_index()
3937

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

0 commit comments

Comments
 (0)